[Python知识库]【python学习】np.array数据保存与访问

保存一份数据(np.save)

import numpy as np
list1=[1,2]
np.save('save_exam0',list1)
npzfile0=np.load('save_exam0.npy') #保存一份数据
print(npzfile0)#此处调用数据不能用默认名称arr_0,只能使用列表的访问顺序

输出结果：[1 2]

保存多份数据(np.savez)

import numpy as np
list5=[1,2]
list6=[3,4]
list7=[5,6]
np.savez('save_exam2',one=list5,two=list6,three=list7)
npzfile2=np.load('save_exam2.npz') #保存多份数据
print(npzfile2['one'],npzfile2['two'],npzfile2['three'])

输出结果：[1 2] [3 4] [5 6]

保存多份数据+函数应用

import numpy as np
def exam(A):
    list=[A,2,3,4,5]
    list2=np.array(list)
    list3=1-list2
    list4=6-list2
    return list3,list4
list3,list4=exam(1)
np.savez('save_exam',list3,list4)#保存多份数据
npzfile=np.load('save_exam.npz') 
print(npzfile['arr_0'],npzfile['arr_1'])

输出结果：[ 0 -1 -2 -3 -4] [5 4 3 2 1]