当遍历NumPy数组时出现错误,可能是由于以下原因之一:
import numpy as np
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
for i in range(arr.shape[0]):
for j in range(arr.shape[1]):
print(arr[i, j])
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
for element in arr:
print(element)
正确的方法是使用索引或切片进行遍历:
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
for i in range(arr.shape[0]):
print(arr[i])
或者使用NumPy提供的函数进行遍历,例如:
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
np.nditer(arr)
for element in np.nditer(arr):
print(element)
确保遵循上述解决方法之一,以正确遍历NumPy数组。
上一篇:遍历numpy数组的最后维度