numpy的數組遍歷錯誤：IndexError：索引1超出範圍爲0軸與大小1

Question

numpy的具有形狀（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函數計算應該返回正確的行和列的值。