numpy数组的遍历
时间: 2024-10-29 18:12:12 浏览: 19
Python数据分析03:Numpy数组和通用函数
在NumPy中,可以使用`np.nditer()`函数来遍历数组。这里有两个不同的示例:
**第一个示例[^1]**展示了如何遍历数组并打印每个元素,但不改变原始数组的内容:
```python
import numpy as np
a = np.arange(0, 60, 5).reshape(3, 4)
print("原数组", a)
for x in np.nditer(a, flags=['external_loop'], order='F'):
print(x)
```
在这个例子中,`flags=['external_loop']`用于保持外部循环(即,不改变数组),`order='F'`指定按列优先(Fortran风格)顺序遍历。
**第二个示例**则演示了如何遍历数组的同时更新数组元素,这里使用`op_flags=['readwrite']`标志允许读写操作:
```python
import numpy as np
a = np.arange(0, 60, 5).reshape(3, 4)
print("原数组是:", a)
for x in np.nditer(a, op_flags=['readwrite']):
x[...] = 2 * x
print('修改后的数组是:', a)
```
在这个例子中,数组的每个元素都被乘以2,因此遍历完成后得到的是修改后的数组。
阅读全文