numpy学习(五)——文件的保存和读写(np.save()、np.load()、np.savez()、np.savetxt()、np.loadtxt())
(1)np.save()和np.load()
#存储数组数据, .npy文件
import numpy as np
import os
os.chdir(r'C:\python数据分析')
ar = np.random.rand(5,5)
print(ar)
np.save('arraytest.npy',ar)#如果文件路径末尾没有扩展名.npy,该扩展名会被自动加上。
#也可以直接np.save(r'C:\python数据分析\arraytest.npy',ar)
#读取数组数据, .npy文件
ar_load = np.load('arraytest.npy')
print(ar_load) #也可以直接np.load(r'C:\python数据分析\arraytest.npy')
(2)np.savez()和np.load()
ar1 = np.random.rand(2,3)
ar2 = np.arange(4)
np.savez(r'C:\python数据分析\arraytest1.npz',ar1,ar2)
r = np.load(r'C:\python数据分析\arraytest1.npz')
print(r) print(r['arr_0'])
(3)np.savetxt()和np.loadtxt()
ar1 = np.random.rand(2,3)
np.savetxt(r'C:\python数据分析\arraytest2.txt',ar1,delimiter=',') #写入的时候指定逗号分割,则读取的时候也要指定逗号分割
ar1_load = np.loadtxt(r'C:\python数据分析\arraytest2.txt',delimiter=',')#指定逗号分割符
print(ar1_load) print(ar1_load.dtype)
#.csv格式 ar1 = np.random.rand(2,3)
np.savetxt(r'C:\python数据分析\arraytest2.csv',ar1,delimiter=',')#csv一定时逗号分隔符
ar1 = np.random.rand(2,3)
np.savetxt(r'C:\python数据分析\arraytest2.txt',ar1,fmt='%.2f')#使用默认分割符(空格),保留两位小数
print(ar1)
ar1_load = np.loadtxt(r'C:\python数据分析\arraytest2.txt')
print(ar1_load)
print(ar1_load.dtype)