在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]])

以上这篇在python中只选取列表中某一纵列的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。

以上是 在python中只选取列表中某一纵列的方法 的全部内容, 来源链接: utcz.com/z/332942.html

回到顶部