python 保存数组和字符串
更新了保存和读取多维数组(narray),保存成.npz
格式
2020.09.19
如果只保存数组是很简单的,但是同时保存数组和字符串有些API用不了,奇怪,也是自己水平不够,解决不了。
先上代码(保存数组和字符串):
我将两个数组和一个维数相同的字符串拼接在一起,然后保存起来
y1=['yes','n1','n2','n3','yes1']
y1 = np.array(y1)
y1 = y1.reshape(5,1)
#print(y1.shape)
x = np.random.rand(5,10)
x1 = np.random.rand(5,10)
z1 = np.concatenate((x1,y1,x),axis=1)
z1 = np.array(z1)
np.save("/home/will/face_train",z1)
保存的格式为.npy
,在文件名后面加不加.npy
都一样,不加就自动补上。
其实一开始想用np.savetxt('PATH',a)
保存成.txt
格式的,但是会报错,字符串类型的问题,估计是因为含有两种类型的数据,如下:
TypeError: Mismatch between array dtype ('<U32') and format specifier
没有找到解决方法 TAT,如果哪位路过的大神,知道的,忘告知,当然不用np.savetxt()
用np.save()
也是没问题的,读取.npy
文件的代码如下:
A=np.load('/home/will/face_train.npy')
print('A.shape:',A.shape)
在这里就要记得加上文件的类型.npy
,不然打不开的
output:
(5, 21)
2020.09.19更新部分
保存和读取多维数组——narray
用np.savez('file_name',data)
,还可以保存多个多维数组np.savez('file_name',data1,data2,data3)
For example:
data1=np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
data2=np.array([1, 2, 3], [4, 5, 6])
data3=np.array([1, 2, 3])
np.savez('test_data', data1, data2, data3)
读取用np.load('file_name.npz')
注意:读取完还查看不了。
- 需要先用
data = np.load('file_name.npz')
data.files # 查看保存了多少组不同的数组
得到输出:
保存一个多维数组:
['arr_0']
保存三个多维数组:
['arr_0','arr_1','arr_2']
- 之后再通过上面得到的索引号 索引对应数组:
da = data['arr_0']
For example:
data = np.load('test_data.npz')
data.files
# ['arr_0','arr_1','arr_2']
data['arr_0']
# array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
data['arr_1']
# array([1, 2, 3], [4, 5, 6])
data['arr_2']
# array([1, 2, 3])