1. 程式人生 > 程式設計 >在Python中利用pickle儲存變數的例項

在Python中利用pickle儲存變數的例項

在工作中出於某些原因,我們可能需要將變數儲存下來,這樣下次就可以直接去賦值而不用重新執行某些重複耗時的操作了,這裡我們用到了Python的pickle包來做變數的儲存和變數載入,大家注意這個包是python自帶的,不需要另外再去安裝。

pickle用法如下:

pickle.dump(obj,file,protocol=None,*,fix_imports=True)

pickle.load(file,fix_imports=True,encoding=”ASCII”. errors=”strict”)

需要注意這裡的file必須要是以二進位制的形式進行操作(寫入或讀取)

import pickle
c=[1,2,3,4,5,6,7]
a=c*4
f=open('D:\\pycm\\kaggle\\data\\test.txt','wb')
pickle.dump(a,f)
f.close()
print(a)

結果輸出如下:

載入儲存的變數方法如下:

import pickle
f=open('D:\\pycm\\kaggle\\data\\test.txt','rb')
d=pickle.load(f)
f.close()
print(d)

結果輸出如下:

這樣就完成變數的儲存和載入操作啦,完整寫法如下:

import pickle
def save_variable(v,filename):
 f=open(filename,'wb')
 pickle.dump(v,f)
 f.close()
 return filename
 
def load_variavle(filename):
 f=open(filename,'rb')
 r=pickle.load(f)
 f.close()
 return r
 
if __name__=='__main__':
 c = [1,7]
 filename=save_variable(c,'D:\\test.txt')
 d=load_variavle(filename)
 print(d==c)

結果輸出如下:

以上這篇在Python中利用pickle儲存變數的例項就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。