使用python自带的shelve
模块,可以作为轻量级的键值数据库,在使用时可以像字典一样使用:
使用shelve
模块的流程如下:
示例程序
import pandas as pd
import shelve
import numpy as np
def main():
_shelve_file = "shelve_file"
shelve_db = shelve.open(_shelve_file)
# 增加
shelve_db['info_a'] = pd.DataFrame(np.random.random(size=(3, 3)))
shelve_db['info_b'] = "value_a"
shelve_db['info_c'] = 2.2
# 删除
del shelve_db['info_c']
# 修改
shelve_db['info_b'] = 1
# 查询
print(shelve_db.get("info_a"))
print(shelve_db.get("info_b"))
print(shelve_db.get("info_c", "info_c数据不存在"))
shelve_db.close() # 主动关闭
if __name__ == '__main__':
main()
在保存后,会在本地添加三个文件,作为这个的保存文件内容
需要注意一下几部分内容:
- 在使用结束后,需要显性的调用
.close()
- 在使用打开
open()
的时候有四个参数:- c(默认):打开数据库进行读写,如果不存在则创建它
- r:只读
- w:读写模式
- n:始终创建一个新的数据库