NumPy数组可以以多种方式进行迭代,包括使用for循环、nditer函数和flat属性。
- 使用for循环迭代:
import numpy as np arr = np.array([1, 2, 3, 4, 5]) for x in arr: print(x)
- 使用nditer函数进行迭代:
import numpy as np arr = np.array([[1, 2], [3, 4]]) for x in np.nditer(arr): print(x)
- 使用flat属性进行迭代:
import numpy as np arr = np.array([[1, 2], [3, 4]]) for x in arr.flat: print(x)
这些方法可以用来迭代多维数组中的元素,可以根据具体需求选择合适的方法进行迭代。