Python 取出一列

list取出一列不能直接切片

>>> a=[[1,2,3], [4,5,6]]
>>> a[:, 0]                   # 尝试用数组的方法读取一列失败
TypeError: list indices must be integers or slices, not tuple

需要使用列表解析的方法来读取一列

>>> b = [i[0] for i in a]     # 从a中的每一行取第一个元素。
>>> print(b)
[1, 4]

想要切片取出一列的话，可以先将list转为numpy.array，就可以了