numpy多维数组的遍历
时间: 2024-10-29 21:12:13 浏览: 23
NumPy基础:多维数组
5星 · 资源好评率100%
Numpy 提供了高效的数据结构 `ndarray` 来支持多维数组遍历[^1]。对于多维数组,可以使用以下几种方式:
1. 使用索引访问元素:
```python
import numpy as np
array_2d = np.arange(0, 10).reshape(2, 5)
for row in array_2d:
for elem in row:
print(elem, end=" ")
print() # 换行
```
2. 利用`.T`属性进行转置遍历:
```python
print(array_2d.T) # 打印列向量
for column in array_2d.T:
print(column)
```
3. 如果想遍历整个数组,可以使用内置的迭代器(如`itertools.product`)[^2],但这通常用于更复杂的组合操作而非常规遍历:
```python
from itertools import product
for i, j in product(range(array_2d.shape[0]), range(array_2d.shape[1])):
print(array_2d[i][j], end=" ")
```
阅读全文