NumPy数组迭代的方法是什么

681
2024/6/7 12:33:11
栏目: 编程语言
开发者测试专用服务器限时活动,0元免费领,库存有限,领完即止! 点击查看>>

NumPy数组可以以多种方式进行迭代,包括使用for循环、nditer函数和flat属性。

  1. 使用for循环迭代:
import numpy as np

arr = np.array([1, 2, 3, 4, 5])

for x in arr:
    print(x)
  1. 使用nditer函数进行迭代:
import numpy as np

arr = np.array([[1, 2], [3, 4]])

for x in np.nditer(arr):
    print(x)
  1. 使用flat属性进行迭代:
import numpy as np

arr = np.array([[1, 2], [3, 4]])

for x in arr.flat:
    print(x)

这些方法可以用来迭代多维数组中的元素,可以根据具体需求选择合适的方法进行迭代。

辰迅云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>

推荐阅读: NumPy如何改变数组的形状