菜单
×
   ❮     
HTML CSS JAVASCRIPT SQL PYTHON JAVA PHP 如何 W3.CSS C C++ C# BOOTSTRAP REACT MYSQL JQUERY EXCEL XML DJANGO NUMPY PANDAS NODEJS R TYPESCRIPT ANGULAR GIT POSTGRESQL MONGODB ASP AI GO KOTLIN SASS VUE DSA GEN AI SCIPY AWS 网络安全 数据科学
     ❯   

NumPy 数组迭代


迭代数组

迭代意味着逐个遍历元素。

由于我们在 numpy 中处理多维数组,我们可以使用基本的 Python for 循环来完成此操作。

如果我们对一维数组进行迭代,它将逐个遍历每个元素。

示例

迭代以下一维数组的元素

import numpy as np

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

for x in arr
  print(x)
自己动手试一试 »

迭代二维数组

在二维数组中,它将遍历所有行。

示例

迭代以下二维数组的元素

import numpy as np

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

for x in arr
  print(x)
自己动手试一试 »

如果我们对 *n* 维数组进行迭代,它将逐个遍历第 n-1 维。

要返回实际值(标量),我们必须在每个维度上迭代数组。

示例

迭代二维数组的每个标量元素

import numpy as np

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

for x in arr
  for y in x
    print(y)
自己动手试一试 »


迭代三维数组

在三维数组中,它将遍历所有二维数组。

示例

迭代以下三维数组的元素

import numpy as np

arr = np.array([[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]])

for x in arr
  print(x)
自己动手试一试 »

要返回实际值(标量),我们必须在每个维度上迭代数组。

示例

迭代到标量

import numpy as np

arr = np.array([[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]])

for x in arr
  for y in x
    for z in y
      print(z)
自己动手试一试 »

使用 nditer() 迭代数组

函数 nditer() 是一个辅助函数,可用于从非常基本到非常高级的迭代。它解决了一些我们在迭代中遇到的基本问题,让我们通过示例来了解它。

迭代每个标量元素

在基本的 for 循环中,迭代数组的每个标量需要使用 *n* 个 for 循环,这对于维度非常高的数组来说很难编写。

示例

迭代以下三维数组

import numpy as np

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

for x in np.nditer(arr)
  print(x)
自己动手试一试 »

迭代具有不同数据类型的数组

我们可以使用 op_dtypes 参数并传入预期的数据类型,以便在迭代时更改元素的数据类型。

NumPy 不会就地更改元素的数据类型(即元素在数组中的位置),因此它需要一些其他空间来执行此操作,这个额外空间称为缓冲区。为了在 nditer() 中启用它,我们传入 flags=['buffered']

示例

将数组作为字符串迭代

import numpy as np

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

for x in np.nditer(arr, flags=['buffered'], op_dtypes=['S'])
  print(x)
自己动手试一试 »

以不同的步长迭代

我们可以使用筛选器,然后进行迭代。

示例

迭代二维数组的每个标量元素,跳过 1 个元素

import numpy as np

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

for x in np.nditer(arr[:, ::2])
  print(x)
自己动手试一试 »

使用 ndenumerate() 进行枚举迭代

枚举意味着逐个提及事物的序列号。

有时我们在迭代时需要元素的相应索引,ndenumerate() 方法可用于这些用例。

示例

枚举以下一维数组的元素

import numpy as np

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

for idx, x in np.ndenumerate(arr)
  print(idx, x)
自己动手试一试 »

示例

枚举以下二维数组的元素

import numpy as np

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

for idx, x in np.ndenumerate(arr)
  print(idx, x)
自己动手试一试 »


×

联系销售

如果您想将 W3Schools 服务用于教育机构、团队或企业,请发送电子邮件给我们
sales@w3schools.com

报告错误

如果您想报告错误,或想提出建议,请发送电子邮件给我们
help@w3schools.com

W3Schools 经过优化,旨在方便学习和培训。示例可能经过简化,以提高阅读和学习体验。教程、参考资料和示例会不断审查,以避免错误,但我们无法保证所有内容的完全正确性。使用 W3Schools 即表示您已阅读并接受我们的使用条款Cookie 和隐私政策

版权所有 1999-2024 Refsnes Data。保留所有权利。W3Schools 由 W3.CSS 提供支持