【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')