numpy保存与读取

numpy保存与读取

一、保存默认的npy,npz二进制格式

import numpy as np
arr_1 = np.arange(20).reshape(4,5)
#保存数组的二进制文件，保存一个,保存的文件名字为'a',(默认保存一个的文件格式为npy)
np.save('a',arr_1)

#%%%%%%%%%%%%%%%%%%%%%%%%%%%

#读取数组的文件
arr_2 = np.load('a.npy')
print(arr_2)

# 保存多个，默认的文件格式为‘.npz’
np.savez('text',nd_1=arr_1,nd_2=arr_2)
#读取文件
arr_4 = np.load('text.npz')
print(arr_4['nd_1'])

二、指定文件格式储存

arr_3 = np.array([
['col_1','col_2','col_3'],
['java','python','go'],
['mysql','redis','mongodb']
])
print(arr_3)
np.savetxt('aaa.csv',arr_3,delimiter=',',fmt='%s')

#指定保存文件名与格式为：‘aaa.csv’,每个数据用逗号隔开，以'%s'保存
delimiter:分隔符，csv文件需要指定','
fmt指定的时数据的格式 '%s' '%d' '%f'小数

读取‘aaa.csv’文件，用逗号隔开，数据类型为字符串类型，usecols是指要读取的列，下标从0开始，不过一定要是一个可迭代对象，比如：[1]，(1,)，记住如果是元组且只有一个元素，一定要加一个逗号：
arr_4=np.genfromtxt(
    'aaa.csv',delimiter=',',dtype=str,usecols=(1,)
)
print(arr_4)