本文介绍了将numpy数组保存在mongodb中的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!
问题描述
我有几个MongoDB文档,其中一个字段最好表示为矩阵(numpy数组).我想将此文档保存到MongoDB,该怎么做?
I have a couple of MongoDB documents wherein one my the fields is best represented as a matrix (numpy array). I would like to save this document to MongoDB, how do I do this?
{
'name' : 'subject1',
'image_name' : 'blah/foo.png',
'feature1' : np.array(...)
}
推荐答案
对于一维numpy数组,可以使用列表:
For a 1D numpy array, you can use lists:
# serialize 1D array x
record['feature1'] = x.tolist()
# deserialize 1D array x
x = np.fromiter( record['feature1'] )
对于多维数据,我相信您需要使用pickle和pymongo.binary.Binary:
For multidimensional data, I believe you'll need to use pickle and pymongo.binary.Binary:
# serialize 2D array y
record['feature2'] = pymongo.binary.Binary( pickle.dumps( y, protocol=2) ) )
# deserialize 2D array y
y = pickle.loads( record['feature2'] )
这篇关于将numpy数组保存在mongodb中的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持!