Numpy—— 保存和导入文件

Posted 2020-12-10 long5683

一、文件读写

Numpy可以方便的进行文件读写，如下面这种格式的文本文件：

# 使用np.fromfile从文本文件‘housing.data‘读入数据
# 这里要设置参数sep = ‘ ‘，表示使用空白字符来分隔数据
# 空格或者回车都属于空白字符，读入的数据被转化成1维数组
d = np.fromfile(‘./work/housing.data‘, sep = ‘ ‘)

二、文件保存

Numpy提供了save和load接口，直接将数组保存成文件(保存为.npy格式)，或者从.npy文件中读取数组。

# 产生随机数组a
a = np.random.rand(3,3)
np.save(‘a.npy‘, a)

# 从磁盘文件‘a.npy‘读入数组
b = np.load(‘a.npy‘)

# 检查a和b的数值是否一样
check = (a == b).all()
check

True