numpy——文件读取与保存

numpy中的文件读取与保存

import numpy as np
import pickle

读取

1.dst = numpy.load(file, mmap_mode=None, allow_pickle=False, fix_imports=True, encoding=‘ASCII’)

• file：文件名或者路径加文件名或者文件对象
• allow_pickle：为True时，可以从将pkl保存的文件读入，但是这样不好，一般默认为False

f = open("a", mode="rb")
# 用文件对象
a = np.load(f, allow_pickle=True)
a

{'age': 22, 'sex': 'male', 'height': 180}

# 用文件名
x = np.load("q.npy")
x

array([1, 2, 3])

保存

numpy.save(file, arr, allow_pickle=True)

• file：可以是文件名或者路径名加文件名，或者是文件对象；如果没有给出文件的后缀，则默认保存为*.npy文件
• arr：要保存的数组
• allow_pickle：一般不管这个参数，最好不要混合使用两种数据

# 直接用文件名保存数组
a = np.array([1, 2, 3])
np.save("q", a)
# 用文件对象保存
f = open("arr", mode="wb")
np.save(f, a)

numpy.savez(file, *args, **kwds)

• file：文件名或者路径加文件名，或者是文件对象；如果没有指定文件后缀，则默认保存为*.npz
• 可以传入多个数组对象，用逗号隔开就行

numpy.savez()会将保存的多个数组以字典形式保存，每个数组都会默认给一个key，从“arr_0”开始。也可以在传入多个数组时用关键字自己指定其key值。通过obj = np.load（npz文件）加载后，可以通过obj.files属性查看所有保存数组的key值，类似于dict.keys。并且可以通过key值索引保存的数组。

j = np.array([1, 5, 8])
k = np.array([11, 31, 9])
l = np.array([67, 40, 23])
# 用默认key值
np.savez("arr3",  j, k, l)

# 要写出完整的文件名，包括后缀
arr3 = np.load("arr3.npz")
print(arr3.files)
print(arr3["arr_0"])

['arr_0', 'arr_1', 'arr_2']
[1 5 8]

# 指定key值
np.savez("arr3_d", embed=j, ot=k, qq=l)

arr3_d = np.load("arr3_d.npz")
arr3_d.files

['embed', 'ot', 'qq']