numpy数组遍历错误：IndexError：索引1超出了0的大小为1的范围

Question

numpy with shape（5,1） 具有以下要素：

[[1]
 [2]
 [3]
 [4]
 [5]]

你如何遍历和打印每个元素？ 如下：

1
2
3
4
5

尝试

for row in range(N):
    for col in range(D):
        print(input_array[row][col])

错误

Error: IndexError: index 1 is out of bounds for axis 0 with size 1

Answer 1

您的N, D值必定是错误的。 <或者

N, D = input_array.shape

继续使用您的代码，或直接

for row in input_array:
    for token in row:
        print(token)

Answer 2

根据错误，N值是错误的。它应该是[0]。

使用numpy.ndarray.shape函数计算应该返回正确的行和列值。