设置allow_pickle=True
时间: 2023-07-15 10:12:50 浏览: 1208
keras_pickle_wrapper:一个包装Keras模型以使其腌制的小型库
在使用numpy.load()函数加载.npy或者.npz文件时,如果文件中保存了Python对象(比如列表、字典等),需要设置allow_pickle=True,才能够正确地将这些对象加载到内存中。这是因为默认情况下,numpy.load()函数会禁止加载pickle格式的数据,以防止恶意代码的执行。因此,如果你需要加载包含Python对象的.npy或者.npz文件,需要显式地设置allow_pickle参数为True,示例如下:
```python
import numpy as np
# 加载包含Python对象的.npy文件
data = np.load('data.npy', allow_pickle=True)
# 加载包含Python对象的.npz文件
npzfile = np.load('data.npz', allow_pickle=True)
```
需要注意的是,如果你不确定.npy或者.npz文件中是否包含Python对象,建议在加载数据时都设置allow_pickle=True,这样可以避免出现错误。
阅读全文