如果是1维或2维数组,可以使用np.savetxt()
, 如:
import numpy as np# 创建数组(2维)a = np.arange(100).reshape((20, 5))# 写入文件np.savetxt(fname="data.csv", X=a, fmt="%d",delimiter=",")# 读取文件b = np.loadtxt(fname="data.csv", dtype=np.int, delimiter=",")
如果是高维数组,可以使用np.save()
,如:
import numpy as np# 创建数组(3维)a = np.arange(100).reshape((10, 5, 2))# 存储np.save(file="data.npy", arr=a)# 读取b= np.load(file="data.npy")