python存储变量_将python 中的变量保存到本地,以及删除本地变量。

如何将python中的变量保存在本地?

将python 的一些代码保存在本地, 特别是一些需要大量运算的结果,例如 机器学习里面的模型,,放在本地,还是比较好用的。下次就可以直接拿出来使用就好。

然后现在添加了一个删除变量的小功能。

见代码:

import shelve

from contextlib import closing

class local_cache:

def __init__(self, cache='var.pkl'):

self.cache = cache

def __setitem__(self, key, value):

"""

key: 变量名

value: 变量值

cache: 缓存名

"""

with closing(shelve.open(self.cache, 'c')) as shelf:

shelf[key] = value

def __getitem__(self, key):

"""

key : 变量名

return:变量值

"""

with closing(shelve.open(self.cache, 'r')) as shelf:

return shelf.get(key)

def remove(self, key):

"""

key: 变量名

如果 变量存在则删除, 如果不存在,会抛出一个异常,由调用方去处理

"""

with closing(shelve.open(self.cache, 'r')) as shelf:

del shelf[key]

return True

if __name__ == '__main__':

# 保存变量和提取变量

# cache = local_cache('var.pkl')

# cache['ok'] = 'okk'

# ok = cache['ok']

# print(ok)

# 删除变量

cache = local_cache('var.pkl')

cache.remove('ok')

ok = cache['ok']

print(ok)

本地保存文件:

在redis中保存python对象: xxxxxx

你可能感兴趣的:(python存储变量)