從numpy數組中提取列

Question

假設我有一個numpy數組

A = np.array([[1,2,3,4],[5,6,7,8]])

我想按行訪問它

for row in A:
  print(row)

導致我有

[1 2 3 4]
[5 6 7 8]

是否有類似的列明智的方法來訪問數組，這將導致我有

[1 5]
[2 6]
[3 7]
[4 8]

我知道我可以使用索引，但是，我只想知道是否有一種方法可以在沒有索引的情況下明智地訪問數組列。

謝謝

Answer 1

轉置數組應該可以為您提供所需的內容：

for item in A.T:
    print(item)

T屬性是transpose()方法的縮寫，並返回數組上的視圖。

Answer 2

您可以選擇A的第i列，如下所示：

for i in range(A.shape[1]):
    print(A[:, i])