[英]SQLAlchemy - filtering func.count within query
假设我有一个带有列的表,该表具有一些整数值,并且我想计算该列的值超过200的百分比。
这是一个关键,如果我可以在可以使用group_by的一个查询中执行此操作,则希望这样做。
results = db.session.query(
ClassA.some_variable,
label('entries', func.count(ClassA.some_variable)),
label('percent', *no clue*)
).filter(ClassA.value.isnot(None)).group_by(ClassA.some_variable)
或者,也可以考虑不喜欢在客户端进行百分比计算,就像这样。
results = db.session.query(
ClassA.some_variable,
label('entries', func.count(ClassA.some_variable)),
label('total_count', func.count(ClassA.value)),
label('over_200_count', func.count(ClassA.value > 200)),
).filter(ClassA.value.isnot(None)).group_by(ClassA.some_variable)
但是我显然不能在count statemenet内进行过滤,也无法在查询的末尾应用过滤器,因为如果在末尾应用> 200约束,total_count将无法工作。
也可以使用RAW SQL,它不一定是Sqlalchemy
不幸的是,MariaDB不支持聚合FILTER子句 ,但是您可以使用CASE表达式或NULLIF解决该问题,因为COUNT返回给定表达式的非空值的计数:
from sqlalchemy import case
...
func.count(case([(ClassA.value > 200, 1)])).label('over_200_count')
考虑到这一点,您可以简单地计算百分比
(func.count(case([(ClassA.value > 200, 1)])) * 1.0 /
func.count(ClassA.value)).label('percent')
尽管有一个优势:如果func.count(ClassA.value)
为0怎么办? 根据您是将0还是NULL视为有效的返回值,您可以使用另一个CASE表达式或NULLIF:
dividend = func.count(case([(ClassA.value > 200, 1)])) * 1.0
divisor = func.count(ClassA.value)
# Zero
case([(divisor == 0, 0)],
else_=dividend / divisor).label('percent')
# NULL
(dividend / func.nullif(divisor, 0)).label('percent')
最后,您可以为mysql方言创建一个编译扩展 ,该扩展将FILTER子句重写为合适的CASE表达式:
from sqlalchemy.ext.compiler import compiles
from sqlalchemy.sql.expression import FunctionFilter
from sqlalchemy.sql.functions import Function
from sqlalchemy import case
@compiles(FunctionFilter, 'mysql')
def compile_functionfilter_mysql(element, compiler, **kwgs):
# Support unary functions only
arg0, = element.func.clauses
new_func = Function(
element.func.name,
case([(element.criterion, arg0)]),
packagenames=element.func.packagenames,
type_=element.func.type,
bind=element.func._bind)
return new_func._compiler_dispatch(compiler, **kwgs)
有了这一点,您可以将股息表示为
dividend = func.count(1).filter(ClassA.value > 200) * 1.0
编译成
In [28]: print(dividend.compile(dialect=mysql.dialect()))
count(CASE WHEN (class_a.value > %s) THEN %s END) * %s
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.