如何将numpy数组保存为.npy文件
时间: 2024-05-04 19:21:41 浏览: 167
你可以使用 `numpy.save()` 方法将一个 numpy 数组保存为 .npy 文件。以下是一个简单的示例:
```python
import numpy as np
# 创建一个 numpy 数组
arr = np.array([1, 2, 3, 4, 5])
# 将数组保存为 .npy 文件
np.save('my_array.npy', arr)
```
这将在当前目录下创建一个名为 `my_array.npy` 的文件,其中包含保存的 numpy 数组。你可以使用 `numpy.load()` 方法加载该文件并恢复 numpy 数组:
```python
# 加载 .npy 文件
loaded_arr = np.load('my_array.npy')
# 打印恢复的数组
print(loaded_arr)
```
输出:
```
[1 2 3 4 5]
```
如果您要保存多个 numpy 数组到同一个文件中,您可以使用 `numpy.savez()` 方法,它允许您将多个数组保存在一个压缩文件中。例如:
```python
# 创建两个 numpy 数组
a = np.array([1, 2, 3])
b = np.array([4, 5, 6])
# 将两个数组保存在一个 .npz 文件中
np.savez('my_arrays.npz', a=a, b=b)
```
这将在当前目录下创建一个名为 `my_arrays.npz` 的文件,其中包含保存的两个 numpy 数组。你可以使用 `numpy.load()` 方法加载该文件并恢复 numpy 数组:
```python
# 加载 .npz 文件
loaded_data = np.load('my_arrays.npz')
# 从加载的数据中获取数组
loaded_a = loaded_data['a']
loaded_b = loaded_data['b']
# 打印恢复的数组
print(loaded_a)
print(loaded_b)
```
输出:
```
[1 2 3]
[4 5 6]
```
阅读全文