【python】将变量保存在本地及读取

时间:2024-01-27 17:46:07

在用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')