【python】將變量保存在本地及讀取

• 2020 年 11 月 12 日
• 筆記
• python小技巧

在用jupyter notebook寫python代碼的過程中會產生很多變量，而關閉後或者restart jupyter kernel後所有變量均會消失，想要查看變量就必須將代碼重新再運行一遍，而想在另一個jupyter notebook中調用變量就更加麻煩。在運行時間很長的代碼中將變量保存下來能夠節省很多事。

那就開始吧！

我用到的包是pickle

1.在使用之前首先需要導入包：

import pickle

2.導入包後即可開始實質性操作，我們定義保存變量和讀取變量的函數。

保存變量函數：

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

3.保存變量和讀取變量操作。

保存變量：將變量results保存在results.txt文件中。

filename = save_variable(results, 'results.txt')

讀取變量：從results.txt中讀取變量內容給results 。

results = load_variavle('results.txt')

最後將代碼放在一起，想用哪段用哪段。

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
filename = save_variable(results,'results.txt')
results = load_variavle('results.txt')