请提供一段完整可以运行的代码:把N维数组转化为一维数组
时间: 2023-06-02 17:05:15 浏览: 91
Java编程一维数组转换成二维数组实例代码
5星 · 资源好评率100%
好的,以下是Python的代码示例:
```
import numpy as np
def flatten(arr):
shape = arr.shape
arr = arr.reshape(-1)
return arr
# Test the function
arr = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
print(flatten(arr))
```
这段代码可以将N维数组(在此例中,arr的shape为(2,2,2))转化为一维数组(即把所有元素都放到同一个维度中)。在本例中,输出为[1 2 3 4 5 6 7 8]。
阅读全文