Numpy数组的保存与加载

Posted 2021-06-30 Xavier Jiezou

文章目录

• 单个数组⇔二进制文件
• 单个数组⇔文本文件（TXT）
• 多个数组⇔二进制文件（未压缩）
• 多个数组⇔二进制文件（压缩）

---

单个数组⇔二进制文件

保存：numpy.save(file, arr)
加载：numpy.load(file)

将数组保存为二进制文件。推荐文件后缀.npy。

>>> import numpy as np
>>> with open('test.npy', 'wb') as f:  
...     np.save(f, np.array([1, 2, 3]))
>>> with open('test.npy', 'rb') as f:  
...     a = np.load(f)
>>> a
array([1, 2, 3])

单个数组⇔文本文件（TXT）

保存：numpy.savetxt(fname, X)
加载：numpy.genfromtxt(fname)

将单个数组保存为文本（txt）文件。可以直观地看到保存的数组。

>>> import numpy as np
>>> np.savetxt('test.txt', np.array([1, 2, 3]))
>>> a = np.genfromtxt('test.txt')
>>> a
array([1., 2., 3.])

多个数组⇔二进制文件（未压缩）

保存：numpy.savez(file)
加载：numpy.load(file)

将多个数组以键值对的方式保存为二进制文件。推荐文件后缀.npz。

>>> import numpy as np
>>> a = np.array([1])
>>> b = np.array([2])
>>> c = np.array([3])
>>> with open('test.npz', 'wb') as f:
...     np.savez(f, a=a, b=b, c=c)
>>> with open('test.npz', 'rb') as f:
...     a = np.load(f)['a']
...     b = np.load(f)['b']
...     c = np.load(f)['c']
>>> a
array([1])
>>> b
array([2])
>>> c
array([3])

多个数组⇔二进制文件（压缩）

保存：numpy.savez_compressed(file)
加载：numpy.load(file)

将多个数组以键值对的方式保存为二进制压缩文件（使用的压缩算法是ZIP_DEFLATED）。推荐文件后缀.npz。

>>> import numpy as np
>>> a = np.array([1])
>>> b = np.array([2])
>>> c = np.array([3])
>>> with open('test.npz', 'wb') as f:
...     np.savez_compressed(f, a=a, b=b, c=c)
>>> with open('test.npz', 'rb') as f:
...     a = np.load(f)['a']
...     b = np.load(f)['b']
...     c = np.load(f)['c']
>>> a
array([1])
>>> b
array([2])
>>> c
array([3])