输入和输出（二进制文件、文本文件、文本格式选项）

输入和输出

1.numpy二进制文件

save() 、 savez() 和 load() 函数以 numpy 专用的二进制类型（npy、npz）保存和读取数据，这
三个函数会自动处理ndim、dtype、shape等信息，使用它们读写数组非常方便，但是 save() 输出
的文件很难与其它语言编写的程序兼容。
npy格式：以二进制的方式存储文件，在二进制文件第一行以文本形式保存了数据的元信息（ndim，
dtype，shape等），可以用二进制工具查看内容。
npz格式：以压缩打包的方式存储文件，可以用压缩软件解压。

1.1 numpy.save()

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

1.2 nuumpy.load()

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

import numpy as np
outfile = r'.\test.npy'
np.random.seed(20201123)
x = np.random.uniform(0, 1, [3, 5])
np.save(outfile, x)
y = np.load(outfile)
print(y)

[[0.03911501 0.91357784 0.21820335 0.61869406 0.25371066]
 [0.75731372 0.16270282 0.77498589 0.41520052 0.15138986]
 [0.34765902 0.22682386 0.80095883 0.39216596 0.79913296]]

1.3 numpy.savez()

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

savez() 第一个参数是文件名，其后的参数都是需要保存的数组，也可以使用关键字参数为数组起一个名字，非关键字参数传递的数组会自动起名为 arr_0, arr_1, … 。

savez() 输出的是一个压缩文件（扩展名为npz），其中每个文件都是一个 save() 保存的npy文件，文件名对应于数组名。 load() 自动识别npz文件，并且返回一个类似于字典的对象，可以通过数组名作为关键字获取数组的内容。

【例】将多个数组保存到一个文件，可以使用 numpy.savez() 函数。

import numpy as np  
outfile = r'.\test.npz'
x = np.linspace(0, np.pi, 5) #linspace函数生成（0，Π）中的五个随机数
y = np.sin(x)
z = np.cos(x)
np.savez(outfile, x, y, z_d=z)
data = np.load(outfile)
np.set_printoptions(suppress=True)
print(data.files)

['z_d', 'arr_0', 'arr_1']

print(data['z_d'])

[ 1.          0.70710678  0.         -0.70710678 -1.        ]

print(data['arr_0'])

[0.         0.78539816 1.57079633 2.35619449 3.14159265]

print(data['arr_1'])

[0.         0.70710678 1.         0.70710678 0.        ]

用解压软件打开 test.npz 文件，会发现其中有三个文件： arr_0.npy , arr_1.npy , z_d.npy ，其中分别保存着数组 x,y,z 的内容

2.文本文件

savetxt() ， loadtxt() 和 genfromtxt() 函数用来存储和读取文本文件（如TXT，CSV等）。
genfromtxt() 比 loadtxt() 更加强大，可对缺失数据进行处理

2.1 numpy.savetxt()

• numpy.savetxt(fname, X, fmt=’%.18e’, delimiter=’ ‘, newline=’\n’, header=’’, footer=’’,comments=’# ', encoding=None) Save an array to a text file.

fname：文件路径。

X：存入文件的数组。

fmt：写入文件中每个元素的字符串格式，默认’%.18e’（保留18位小数的浮点数形式）。

delimiter：分割字符串，默认以空格分隔。

2.2 numpy.loadtxt()

• numpy.loadtxt(fname, dtype=float, comments=’#’, delimiter=None, converters=None,
  skiprows=0, usecols=None, unpack=False, ndmin=0, encoding=‘bytes’, max_rows=None) Load
  data from a text file.

fname：文件路径。

dtype：数据类型，默认为float。

comments: 字符串或字符串组成的列表，默认为# ， 表示注释字符集开始的标志。

skiprows：跳过多少行，一般跳过第一行表头。

usecols：元组（元组内数据为列的数值索引）， 用来指定要读取数据的列（第一