You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

如何用RawSQL为Django查询集annotate字段?报错求助

解决Django annotate中RawSQL报错“subquery must return only one column”的问题

问题根源

你之前的SQL语句返回了两列数据(c.id和gdd),但annotate要求RawSQL只能返回单一列的值,用于附加到查询集中的每个模型实例上,这就是报错的核心原因。

解决方案:修改SQL为单列关联子查询

调整SQL语句,仅保留gdd的计算逻辑,同时通过主查询的agriculture_commune.id关联子查询,确保每个commune实例计算对应自身的gdd值:

from django.db.models.expressions import RawSQL

# 修改后的SQL,仅返回gdd计算结果
sql_query = """
    SELECT COALESCE(SUM((cm.temp_min + cm.temp_max) / 2 - %s), 0)
    FROM agriculture_communemeteo cm
    WHERE cm.commune_id = agriculture_commune.id
      AND cm.date BETWEEN %s AND %s
"""

# 执行annotate
communes = communes.annotate(gdd=RawSQL(sql_query, [TBASE, start_date, end_date]))

关键说明

  • 移除了原SQL中的SELECT c.id和GROUP BY c.id:主查询已经按每个commune实例返回数据,子查询通过cm.commune_id = agriculture_commune.id自动匹配当前实例的ID,无需额外分组
  • 子查询仅返回单个计算值,完全符合annotate对RawSQL的要求

可选:用Django ORM替代RawSQL(推荐)

如果使用Django 3.2及以上版本,可通过ORM原生函数实现相同逻辑,更符合Django开发风格:

from django.db.models import F, Sum, ExpressionWrapper, FloatField, Coalesce

communes = communes.annotate(
    # 计算单日均温
    daily_temp_avg=ExpressionWrapper(
        (F('communemeteo__temp_min') + F('communemeteo__temp_max')) / 2,
        output_field=FloatField()
    ),
    # 计算GDD,用Coalesce处理无气象数据的情况
    gdd=Coalesce(
        Sum(
            ExpressionWrapper(F('daily_temp_avg') - TBASE, output_field=FloatField()),
            default=0
        ),
        0
    )
).filter(communemeteo__date__range=(start_date, end_date))

内容的提问来源于stack exchange,提问作者Kirill

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.06.26 20:15:17