numpy 读写 npy npz 文件

import numpy as np

创建数组

a1 = np.arange(0,12).reshape(2,2,3)
a1

array([[[ 0, 1, 2],
[ 3, 4, 5]],

​ [[ 6, 7, 8],
​ [ 9, 10, 11]]])

保存单个数组

np.save(
    "1.npy",  # 写入的文件路径
    a1  # 写入的内容
)

读取单个数组

a2 = np.load("1.npy"  # 读取文件路径
             )
a2

array([[[ 0, 1, 2],
[ 3, 4, 5]],

​ [[ 6, 7, 8],
​ [ 9, 10, 11]]])

创建第二个数组

a3 = np.arange(12,24).reshape(2,2,3)
a3

array([[[12, 13, 14],
[15, 16, 17]],

​ [[18, 19, 20],
​ [21, 22, 23]]])

保存多个数组

np.savez(
    "1.npz",  # 写入的文件路径
    a1=a1,  # 写入的数组名字和内容
    a3=a3,  # 写入的数组名字和内容
)

读取多个数组

a1_a3 =  np.load("1.npz"  # 读取文件路径
             )
a1_a3

<numpy.lib.npyio.NpzFile at 0x1d5b76f5400>

a1_a3["a1"]

array([[[ 0, 1, 2],
[ 3, 4, 5]],

​ [[ 6, 7, 8],
​ [ 9, 10, 11]]])

a1_a3["a3"]

array([[[12, 13, 14],
[15, 16, 17]],

​ [[18, 19, 20],
​ [21, 22, 23]]])