从包含零计数的查询中获取所有月份

import pandas as pd

# 1. 确定时间范围
start_date = '2020-01-01'
end_date = '2021-12-31'

# 2. 生成所有月份列表
date_range = pd.date_range(start=start_date, end=end_date, freq='M')

# 3. 查询结果处理（示例数据）
query_result = [
    {'month': '2020-01', 'count': 10},
    {'month': '2020-03', 'count': 5},
    {'month': '2020-05', 'count': 0},
    {'month': '2020-07', 'count': 8},
    {'month': '2020-09', 'count': 0},
    {'month': '2020-11', 'count': 15},
    {'month': '2021-01', 'count': 12},
    {'month': '2021-03', 'count': 0},
    {'month': '2021-05', 'count': 7},
    {'month': '2021-07', 'count': 9},
    {'month': '2021-09', 'count': 0},
    {'month': '2021-11', 'count': 20}
]

# 提取月份和计数
months = [item['month'] for item in query_result]
counts = [item['count'] for item in query_result]

# 4. 合并结果
result = pd.DataFrame({'month': date_range.strftime('%Y-%m')})
result['count'] = result['month'].apply(lambda x: counts[months.index(x)] if x in months else 0)

print(result)