从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])