Python二維數組按列取元素

Numpy數組可以直接切片，但是普通的Python二維數組不行。

方法一

a = [[1,2],[3,4]]
b = [i[0] for i in a]     # 從a中的每一行取第一個元素。
print(b)
[1, 4]

方法二

a = [[1,2],[3,4]]
b = list(zip(*a))
print(b)
[(1, 3), (2, 4)]