python中只选取列表中某一纵列

最新推荐文章于 2023-06-29 05:15:15 发布

原创最新推荐文章于 2023-06-29 05:15:15 发布 · 1.3w 阅读

3 ·

CC 4.0 BY-SA版权

python 专栏收录该内容

51 篇文章

订阅专栏

本文通过实例演示了如何使用Python生成随机整数数组，并将其转换为列表和矩阵形式。进一步展示了如何从矩阵中选取特定列并再次转换为列表的过程。

>>> a=random.randint(1,6,(5,3))
>>> a
array([[5, 3, 1],
       [5, 5, 1],
       [5, 1, 3],
       [1, 4, 3],
       [5, 1, 2]])
>>> b=a.tolist()
>>> b#选取b列表的前2列
[[5, 3, 1], [5, 5, 1], [5, 1, 3], [1, 4, 3], [5, 1, 2]]
>>> c=mat(b)
>>> c
matrix([[5, 3, 1],
        [5, 5, 1],
        [5, 1, 3],
        [1, 4, 3],
        [5, 1, 2]])
>>> d=c[:,0:2]
>>> d#将d转化回列表即可
matrix([[5, 3],
        [5, 5],
        [5, 1],
        [1, 4],
        [5, 1]])