用python代码迭代高维数组中的每一个元素并且修改值的代码
时间: 2023-06-04 22:07:41 浏览: 152
python 迭代
4星 · 用户满意度95%
可以使用numpy库来迭代高维数组中的每一个元素并修改值。以下是示例代码:
```
import numpy as np
# 创建一个4维数组
arr = np.array([[[[1, 2], [3, 4]], [[5, 6], [7, 8]]], [[[9, 10], [11, 12]], [[13, 14], [15, 16]]]])
# 迭代数组并修改每个元素的值
for i in np.ndindex(arr.shape):
arr[i] += 1
print(arr)
```
这个代码将迭代4维数组中的每一个元素并加1。输出结果如下:
```
array([[[[ 2, 3],
[ 4, 5]],
[[ 6, 7],
[ 8, 9]]],
[[[10, 11],
[12, 13]],
[[14, 15],
[16, 17]]]])
```
这个代码可以扩展到任意维度的高维数组,并且对于每个元素都可以进行任意的修改。
阅读全文