如何使用SQL Alchemy对指定列逐行求和生成布尔掩码？

报错根因 #

• 你调用的sa.func.sum()是SQL的纵向聚合函数，作用是对单列的所有行求和，不符合你「逐行对多列做横向求和」的需求
• 你直接将列列表传入sa.func.sum()，SQLAlchemy无法将ColumnClause类型的列表解析为合法SQL参数，因此抛出类型适配错误

解决方案 #

你需要将可变长度的列拼接为横向逐行求和的SQL表达式，直接比较即可得到布尔掩码，无需额外套case语句，所有逻辑全部在SQL查询阶段执行，符合要求：

from functools import reduce
import operator
import sqlalchemy as sa

# 已知参数
target_num = 6
table = 你的SQLAlchemy Table对象
column_list = [sa.column('col_1'), sa.column('col_2'), sa.column('col_3')]

# 构建逐行多列求和表达式，支持任意长度的列列表
row_wise_sum = reduce(operator.add, column_list)
# 直接比较得到布尔掩码表达式
mask = row_wise_sum == target_num

# 场景1：直接查询布尔掩码结果列表
query = sa.select(mask).select_from(table)
result = db.execute(query).scalars().all()
# 输出示例：[True, False, True]

# 场景2：在where子句中过滤符合条件的行
filtered_query = sa.select(table).where(mask)
filtered_rows = db.execute(filtered_query).all()

补充说明 #

如果必须使用case语句兼容特殊逻辑，可以直接将上述表达式套入case：

case_mask = sa.case(
    [(row_wise_sum == target_num, True)],
    else_ = False
)

该写法和直接比较返回的结果完全一致。

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