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，就可以了