我发现在python中为numpy数组建立索引很奇怪

Question

我对python非常陌生，我正在进行一些数据操作。 我原以为numpy中的索引应该是[row][column]，但这不是我在python中执行时看到的。 下面是一个简单的python行为示例。我不明白为什么最后两个命令的结果相同：

import numpy as num

test_arr = num.array([[1, 2, 3],[4, 5, 6], [7, 8, 9]],dtype=num.float32)

test_arr[0][:]
array([1., 2., 3.], dtype=float32)

test_arr[:][0]
array([1., 2., 3.], dtype=float32)

我希望

test_arr[0][:]
array([1., 2., 3.], dtype=float32)

test_arr[:][0]
array([1., 4., 7.], dtype=float32)

有人能解释一下为什么python会这样做，以及如何获取所有行的第0个索引吗

Answer 1

In  : import numpy as num

In  : test_arr = num.array([[1, 2, 3],[4, 5, 6], [7, 8, 9]],dtype=num.float32)

In  : test_arr[0,:]
Out : [1. 2. 3.]

In  : test_arr[:,0]
Out : [1. 4. 7.]

Answer 2

要获取numpy数组中的列，请使用[:,n]，其中n是您的列号

test_arr[:,0]
array([1., 4., 7.], dtype=float32)