在Python中使用BSON将浮点数组序列化/反序列化为二进制文件

Question

我习惯于使用JSON和Numpy在python中存储数组，列表和字典，但是我想使用BSON，因为浮点数只占用4个字节，因此减少文件大小。

使用JSON，我做了以下内容：

import numpy 
import json 

a = numpy.random.rand(12).reshape((3,4)) 

with open('out.json', 'w') as out: 
    json.dump(a.tolist(), out) 

with open('out.json') as inp: 
    b = numpy.array(json.load(inp)) 

print b 

我没有找到一个明显的方式做同样的BSON。我尝试这样做：

import numpy 
from bson import BSON 

a = numpy.random.rand(12).reshape((3,4)) 

b = BSON.encode({'a': a.tolist()}) 

with open('out.bson', 'wb') as out: 
    out.write(b) 

with open('out.bson', 'rb') as inp: 
    print BSON().decode(inp.read()) 

但得到这个错误：

Traceback (most recent call last): 
    File "apaga.py", line 12, in <module> 
    print BSON().decode(inp.read()) 
    File "/usr/lib/python2.7/dist-packages/bson/__init__.py", line 539, in decode 
    (document, _) = _bson_to_dict(self, as_class, tz_aware) 
bson.errors.InvalidBSON: not enough data for a BSON document 

Answer 1

我已经安装了，如图所以也许我使用的是不同的版本将不会导入BSON的版本。要查看导入后的Python类型的帮助（BSON）你的文档......

像这样的东西应该工作：

import bson 
>>> a = numpy.random.rand(12).reshape((3,4)) 
>>> b = bson.dumps({'a':a.tolist()}) 
>>> print bson.loads(b) 
{u'a': [[0.033390565943162254, 0.7282666963459123, 0.03719924011978737, 0.2664821209717694], [0.6145164300761253, 0.3662769247564551, 0.5973843055182299, 0.42908933503924207], [0.05901830243140804, 0.31533731904861184, 0.7158207045507905, 0.12686922689849378]]}