Python中使用pickle模块保存和读取本地变量

Python小技巧——将变量保存在本地及读取

在Python编程中，有时我们需要将变量保存到本地文件中，以便在程序运行结束后再次使用。本文将介绍如何使用pickle模块实现这一功能。

保存变量

首先，我们需要导入pickle模块，并定义一个save_variable函数。该函数接受两个参数：要保存的变量v和文件名filename。函数内部，我们打开文件，将变量v保存到文件中，然后关闭文件并返回文件名。

import pickle
def save_variable(v,filename):
  f=open(filename,'wb')
  pickle.dump(v,f)
  f.close()
  return filename

读取变量

接下来，我们定义一个load_variavle函数。该函数接受一个参数：文件名filename。函数内部，我们打开文件，从文件中读取变量，然后关闭文件并返回读取的变量。

def load_variavle(filename):
  f=open(filename,'rb')
  r=pickle.load(f)
  f.close()
  return r

保存和读取变量

最后，我们调用save_variable函数将变量results保存到文件results.txt中，然后调用load_variavle函数从文件results.txt中读取变量results。

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

以上就是Python小技巧——将变量保存在本地及读取的详细内容，更多关于Python小技巧的资料请关注本站其它相关文章！