在mongodb中保存numpy数组

作者: 小先生2830310
来源: 51数据库
2023-01-03

问题描述

我有几个 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'] )

推荐阅读