利用numpy库创建一个10*10的随机数组,将数组重排成5*2*10的三维数组,用tofile函数写入文件,再导入
时间: 2024-12-16 16:25:41 浏览: 4
numpy中三维数组中加入元素后的位置详解
在Python中,你可以使用NumPy库来轻松地创建数组并进行操作。首先,我们需要安装NumPy,如果你还没有安装,可以使用`pip install numpy`命令。
下面是利用NumPy创建10*10的随机数组,并将其转换为5*2*10三维数组的过程:
```python
import numpy as np
# 创建一个10x10的随机浮点数数组
random_array = np.random.rand(10, 10)
# 将二维数组转置并按行堆叠,形成5x2x10的三维数组
reshaped_array = random_array.reshape(5, 2, -1) # 使用-1表示最后一个维度保持原始大小不变
# 写入名为“output.npy”的二进制文件
np.save('output.npy', reshaped_array)
# 现在我们已经将数组保存到文件中。接下来,我们可以从文件中读取它:
# 导入并验证已保存的数组
loaded_array = np.load('output.npy')
print("Loaded array shape:", loaded_array.shape)
```
阅读全文