Python使用pymongo查询MongoDB按条件仅返回指定字段的优化方法

核心优化逻辑 #

通过MongoDB的*投影（Projection）*参数指定返回字段，无需拉取全量文档，直接降低数据传输开销。

优化后代码示例 #

from pymongo import MongoClient

# 初始化MongoDB连接
client = MongoClient("MongoDB连接串")
db = client["你的数据库名称"]
collection = db["region_report"]

# 配置查询条件
target_day_list = ["2024-05-01", "2024-05-02", "2024-05-03"] # 替换为实际日期列表
fixed_brand = "指定品牌值" # 替换为实际固定Brand值

query_filter = {
    "Day": {"$in": target_day_list},
    "Brand": fixed_brand
}

# 投影配置：仅返回State_name，关闭默认返回的_id字段
return_field = {
    "State_name": 1,
    "_id": 0
}

# 执行查询
query_result = collection.find(filter=query_filter, projection=return_field)

# 直接提取State_name结果列表
state_name_list = [item["State_name"] for item in query_result]

规则说明 #

• 投影参数中字段名: 1表示仅返回对应字段，其余字段默认不返回；字段名: 0表示排除对应字段，其余字段默认返回，两种规则不可混用（_id字段除外）。
• 若需要保留文档的_id字段，删除投影配置中的"_id": 0即可。

额外性能优化建议 #

数据量较大时可添加联合索引避免全表扫描，索引仅需创建一次即可永久复用，创建代码如下：

collection.create_index([("Day", 1), ("Brand", 1)])

加索引后查询效率可再提升数倍至数十倍。

内容的提问来源于stack exchange，提问作者Baron