在mongodb中保存numpy数组

原文链接:https://cloud.tencent.com/developer/ask/32026

对于一维numpy数组,可以使用列表:

# serialize 1D array x
record['feature1'] = x.tolist()

# deserialize 1D array x
x = np.fromiter( record['feature1'] )

对于多维数组,你需要使用pickle和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'] )

猜你喜欢

转载自blog.csdn.net/y_f_raquelle/article/details/84135081