从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 proprety是transpose()方法的缩写，并返回数组的视图。

Answer 2

您可以选择A的i列，如下所示：

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