python存储numpy数据为文件_python利用numpy存取文件

NumPy提供了多种存取数组内容的文件操作函数。保存数组数据的文件可以是二进制格式或者文本格式。二进制格式的文件又分为NumPy专用的格式化二进制类型和无格式类型。

numpy格式的文件可以保存为后缀为(.npy/.npz)格式的文件

1. tofile()和fromfile()

tofile()将数组中的数据以二进制格式写进文件

tofile()输出的数据不保存数组形状和元素类型等信息

fromfile()函数读回数据时需要用户指定元素类型，并对数组的形状进行适当的修改

import numpy as np

# 随机生成12个数字并将其有一维转换成3*4的矩阵形式

a = np.arange(12)

print("一维数组:",a)

a.shape = 3,4

print("3*4的矩阵:",a)

# 将数组中的数据以二进制格式写入到文件

a.tofile('a.bin')

# fromfile在读取numpy文件时需要自己指定数据格式,并且原格式并为保存

b1 = np.fromfile('a.bin', dtype=np.float) # 按照float读取数据

b2 = np.fromfile('a.bin', dtype=np.int) # 按照int读取数据

b3 = np.fromfile('a.bin', dtype=np.int32) # 按照int32读取数据

print('float格式b1:{},\nint格式b2:{},\nint32格式b3:{}'.format(b1,b2,b3))

b3.shape = 3,4

prin