Peewee+PostgreSQL查询报错：product列需在GROUP BY或聚合函数中使用

首先咱们拆解下这个报错的核心原因：PostgreSQL在默认的ONLY_FULL_GROUP_BY模式下，要求SELECT子句里的所有非聚合列必须出现在GROUP BY子句中。你的主查询里SELECT了models.Trade.product，但只把models.Trade.state放进了GROUP BY，数据库无法确定每个state对应的多个product值该如何处理，所以抛出了这个错误。

你的需求是获取指定原产国和年份下，各地区（state）出口总值最高的顶级产品，这里提供两种可行的实现方案：

方案一：基于子查询关联（适配你的原思路） #

先通过子查询算出每个state+product的出口总和，接着找出每个state的最大出口总值，最后关联回去拿到对应的产品：

# 第一步：计算每个state+product组合的出口总值
sum_subquery = (
    models.Trade
    .select(
        models.Trade.state,
        models.Trade.product,
        fn.SUM(models.Trade.export_value).alias("total_export")
    )
    .where(
        models.Trade.origin_country == origin_country,
        models.Trade.year == year
    )
    .group_by(models.Trade.state, models.Trade.product)
    .alias("sum_sub")
)

# 第二步：找出每个state的最大出口总值
max_subquery = (
    sum_subquery
    .select(
        sum_subquery.c.state,
        fn.MAX(sum_subquery.c.total_export).alias("max_total")
    )
    .group_by(sum_subquery.c.state)
    .alias("max_sub")
)

# 第三步：关联两个子查询，拿到每个state对应的顶级产品
final_query = (
    sum_subquery
    .select(
        sum_subquery.c.state,
        sum_subquery.c.product,
        sum_subquery.c.total_export
    )
    .join(
        max_subquery,
        on=(
            (sum_subquery.c.state == max_subquery.c.state) &
            (sum_subquery.c.total_export == max_subquery.c.max_total)
        )
    )
)

方案二：使用窗口函数（更简洁高效） #

PostgreSQL支持窗口函数，用ROW_NUMBER()可以给每个state下的产品按出口总值排序，直接取排名第一的产品，代码更简洁：

from peewee import Window

# 定义窗口：按state分组，按出口总值降序排序
window = Window(
    partition_by=[models.Trade.state],
    order_by=[fn.SUM(models.Trade.export_value).desc()]
)

# 先计算每个state+product的总值，再给每行添加排名
ranked_subquery = (
    models.Trade
    .select(
        models.Trade.state,
        models.Trade.product,
        fn.SUM(models.Trade.export_value).alias("total_export"),
        fn.ROW_NUMBER().over(window).alias("rank")
    )
    .where(
        models.Trade.origin_country == origin_country,
        models.Trade.year == year
    )
    .group_by(models.Trade.state, models.Trade.product)
    .alias("ranked_sub")
)

# 过滤出排名第一的产品
final_query = (
    ranked_subquery
    .select(
        ranked_subquery.c.state,
        ranked_subquery.c.product,
        ranked_subquery.c.total_export
    )
    .where(ranked_subquery.c.rank == 1)
)

补充说明原代码的问题 #

你的原主查询中，SELECT了models.Trade.product但GROUP BY仅包含models.Trade.state。PostgreSQL无法确定每个state对应的product应该取哪个值（一个state可能对应多个product），所以必须把product也加入GROUP BY，或者用聚合函数（比如MAX(product)，但这显然不符合你的需求）。上面的两种方案都避开了这个矛盾，直接定位到每个state的顶级产品。

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