np.save()用法
时间: 2023-10-09 21:08:56 浏览: 60
浅谈numpy中np.array()与np.asarray的区别以及.tolist
5星 · 资源好评率100%
`np.save()` 函数用于将数组保存到二进制文件中,该文件可以稍后使用 `np.load()` 加载回来。
函数语法:
```python
numpy.save(file, arr, allow_pickle=True, fix_imports=True)
```
参数说明:
- file:要保存到的文件,可以是文件名字符串,也可以是已打开的文件描述符。
- arr:要保存的数组。
- allow_pickle:可选参数,布尔值,允许使用 Python pickles 保存数组对象。如果为 False,则只能保存数组数据,默认为 True。
- fix_imports:可选参数,布尔值,控制 pickle 是否支持 Python 2 及 Python 3 之间的兼容性。如果为 True,则 pickle 将在读取时尝试将 Python 2 创建的对象转换为 Python 3 对象,默认为 True。
示例:
```python
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
np.save('array', arr)
```
上述代码将数组 `arr` 保存到名为 `array.npy` 的二进制文件中。你可以使用 `np.load()` 函数加载它:
```python
loaded_arr = np.load('array.npy')
print(loaded_arr)
```
输出:
```
[1 2 3 4 5]
```
阅读全文