mongodb如何存numpy数组-SQL-PHP中文网

mongodb如何存numpy数组

舞姬之光

发布： 2025-11-13 18:44:04

原创

258人浏览过

答案：存储NumPy数组需转换格式，小数组可用tolist()转为列表，大数组推荐用np.save序列化为二进制并存dtype和shape，或通过自定义编码器处理类型。

mongodb如何存numpy数组

在使用 MongoDB 存储 NumPy 数组时，由于 MongoDB 原生不支持 NumPy 的 ndarray 类型，需要先将其转换为 MongoDB 可识别的格式。以下是几种常用方法。

1. 转换为 Python 列表存储

最简单的方式是将 NumPy 数组通过 .tolist() 方法转为 Python 列表，MongoDB 可以直接存储列表结构。

示例代码：

import numpy as np
from pymongo import MongoClient
<h1>创建 NumPy 数组</h1><p>arr = np.array([[1, 2], [3, 4]])</p><h1>转为列表并插入 MongoDB</h1><p>client = MongoClient("mongodb://localhost:27017/")
db = client["test_db"]
collection = db["arrays"]</p><p>collection.insert_one({"name": "matrix", "data": arr.tolist()})

登录后复制

读取时再用 np.array() 恢复：

doc = collection.find_one({"name": "matrix"})
restored_arr = np.array(doc["data"])
print(restored_arr)  # 输出: [[1 2] [3 4]]

登录后复制

2. 序列化为二进制（推荐用于大数组）

对于大型数组，转为列表可能效率低且占用空间大。可以使用 numpy.save 和 io.BytesIO 将数组序列化为二进制数据存储。

示例代码：

import numpy as np
import io
from pymongo import MongoClient
<p>arr = np.random.rand(1000, 1000)  # 大数组示例</p>
                    <div class="aritcle_card">
                        <a class="aritcle_card_img" href="/ai/1146">
                            <img src="https://img.php.cn/upload/ai_manual/000/000/000/175680088775482.png" alt="存了个图">
                        </a>
                        <div class="aritcle_card_info">
                            <a href="/ai/1146">存了个图</a>
                            <p>视频图片解析/字幕/剪辑，视频高清保存/图片源图提取</p>
                            <div class="">
                                <img src="/static/images/card_xiazai.png" alt="存了个图">
                                <span>17</span>
                            </div>
                        </div>
                        <a href="/ai/1146" class="aritcle_card_btn">
                            <span>查看详情</span>
                            <img src="/static/images/cardxiayige-3.png" alt="存了个图">
                        </a>
                    </div>
                <h1>转为二进制</h1><p>buffer = io.BytesIO()
np.save(buffer, arr)
binary_data = buffer.getvalue()</p><h1>存入 MongoDB</h1><p>collection.insert_one({
"name": "large_array",
"data": binary_data,
"dtype": str(arr.dtype),
"shape": arr.shape
})

登录后复制

读取时反序列化：

doc = collection.find_one({"name": "large_array"})
loaded_buffer = io.BytesIO(doc["data"])
restored_arr = np.load(loaded_buffer)

登录后复制

注意：需同时保存 dtype 和 shape 信息以便还原（虽然 np.save/np.load 自带这些信息，但显式保存便于调试）。

3. 使用 BSON 编码扩展（可选）

MongoDB 使用 BSON 格式，可通过自定义编码器支持 NumPy 类型。例如使用 bson 模块配合类型转换。

常见做法是在插入前统一处理 NumPy 类型：

def convert_numpy_types(obj):
    if isinstance(obj, np.ndarray):
        return obj.tolist()
    elif isinstance(obj, (np.int64, np.int32)):
        return int(obj)
    elif isinstance(obj, (np.float64, np.float32)):
        return float(obj)
    return obj

登录后复制

然后在插入前递归处理字典数据。

总结建议

小数组用 tolist() 最方便；大数组推荐二进制存储，节省空间且保留精度；生产环境注意字段命名清晰，并考虑查询需求。读写时做好类型恢复逻辑。

基本上就这些，按实际场景选择合适方式即可。

以上就是mongodb如何存numpy数组的详细内容，更多请关注php中文网其它相关文章！