Python 科学计算包 Numpy 6

NumPy IO

Numpy 可以读写磁盘上的文本数据或二进制数据。

NumPy 为 ndarray 对象引入了一个简单的文件格式:npy

npy 文件用于存储重建 ndarray 所需的数据、图形、dtype 和其他信息。

常用的 IO 函数有:

  • load() 和 save() 函数是读写文件数组数据的两个主要函数,默认情况下,数组是以未压缩的原始二进制格式保存在扩展名为 .npy 的文件中。
  • savze() 函数用于将多个数组写入文件,默认情况下,数组是以未压缩的原始二进制格式保存在扩展名为 .npz 的文件中。
  • loadtxt() 和 savetxt() 函数处理正常的文本文件(.txt 等)
import numpy as np

'''
numpy.save()
numpy.save() 函数将数组保存到以 .npy 为扩展名的文件中。

numpy.load()
numpy.load() 函数来读取数据。

numpy.savez
numpy.savez() 函数将多个数组保存到以 npz 为扩展名的文件中。

numpy.savetxt()
numpy.savetxt() 函数是以简单的文本文件格式存储数据。

numpy.loadtxt()
numpy.loadtxt() 函数来获取数据。

'''

a = np.array([1, 2, 3, 4, 5])

# 保存到 outfile.npy 文件上
np.save('outfile.npy', a)

# 保存到 outfile2.npy 文件上,如果文件路径末尾没有扩展名 .npy,该扩展名会被自动加上
np.save('outfile2', a)

b = np.load('outfile.npy')
print(b);
print('\n')

a = np.array([[1,2,3],[4,5,6]])
b = np.arange(0, 1.0, 0.1)
c = np.sin(b)
# c 使用了关键字参数 sin_array
np.savez("runoob.npz", a, b, sin_array = c)
r = np.load("runoob.npz")
print(r.files) # 查看各个数组名称
print('\n')
print(r["arr_0"]) # 数组 a
print('\n')
print(r["arr_1"]) # 数组 b
print('\n')
print(r["sin_array"]) # 数组 c
print('\n')

a = np.array([1, 2, 3, 4, 5])
np.savetxt('out.txt', a)
b = np.loadtxt('out.txt')
print(b)
print('\n')

>>>

[1 2 3 4 5]

['sin_array', 'arr_0', 'arr_1']

[[1 2 3]
[4 5 6]]

[0. 0.1 0.2 0.3 0.4 0.5 0.6 0.7 0.8 0.9]

[0. 0.09983342 0.19866933 0.29552021 0.38941834
0.47942554
0.56464247 0.64421769 0.71735609 0.78332691]

[1. 2. 3. 4. 5.]