欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

numpy保存与读取

程序员文章站 2022-05-18 19:15:57
...

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)