9. 初识数据存储与读取shelve函数

很多情况下，我们需要读取和写入少量的数据（远小于GB的量级）。在Matlab中采用save和load 两个函数，能够轻松的将变量空间的值存入到文件，也可轻松的将数据值从文件读取到当前活动workspace （类似与python的作用域 scope），我们不用去关注存储方式（二进制或者ascii码）。 Python有这样的函数么？答案你肯定猜到了！他就是shelve模块。

• 如何打开文件和关闭文件？
• 如何存储和读取数据？

如何打开文件？

import shelve
s = shelve.open(filename, flag='c', protocol=None, writeback=False)

解释函数的输入参数之前，回顾下python作用域中的变量是以何种方式存储的？字典（及其类似物,见下面代码）。那么返回值s也就是一个类似的字典 (此处为什么不用f表示而用s呢？可以想想)。

>>> x = 10
>>> vars()
{ 'x': 10, '__name__': '__main__', '__spec__': None,...}

• filename 就不需要解释了吧
• 参数flag表明的是文件的读写权限，下表内容来自'官网'

• protocal（协议，规则） 是什么鬼？肯定是和存储相关的。参数protocol是序列化模式，默认值为0，表示以文本的形式序列化。protocol的值还可以是1或2，表示以二进制的形式序列化。

关闭文件的方式很简单

s.close

为了保证文件顺利关闭，当然也有复杂的方式（try：finally：语句和 with 语句）

>>> s = shelve.open('test.dat')
>>> dir(s)
[ '__exit__', '__ge__', '__getattribute__', '__getitem__', '__gt__', '__hash__', '__init__', '__iter__', 'close', 'dict', 'get', 'items', 'keyencoding', 'keys', 'pop', 'popitem',...]

具有魔法方法__exit__的对象都可以使用设备上下文的方式关闭。

如何存储和读取数据？

如果文件test.dat中没有写入任何东西， 你是读取不到任何数据的。
那么就有必要写入数据，记住，shelve是基于字典的操作。
所以，写入数据很简单。

>>> s = shelve.open('test.dat')
>>> len(s) # 空的文件没有存储数据
0
>>> s['x'] = 'flash'
>>> s['y'] = [1,2,3]
>>> s.close()
>>> s2 = shelve.open('test.dat','r')
>>> len(s2)
2
>>> s2['x']
'flash'
>>> s2['y']
[1, 2, 3]

最终的代码

>>> with shelve.open('test.dat') as s: #使用with 语句就不需要s.close()操作了。
...     for iter in s:
...         print(iter,s[iter])
... 
x flash
y [1, 2, 3]