python - 带有 peewee 的多个聚合函数
问题描述
我几乎不知道 SQL,但我一直在使用peewee
,到目前为止它运行得非常好。我遇到了一个不知道如何解决的问题。这里是设置。
我有两张桌子A
和B
. A
,在其他字段中,有多个B
带有外键。B
有 3 个字段(在我的例子中是 5 个,但最少要复制 3 个):
A:
id param_1 ...
0 ...
1 ...
2 ...
和
B:
id A_key x y z
0 0 3. 5. 1
1 0 4. 4. 2
2 1 2. 3. 3
3 1 5. 1. 4
我想获得与z
最小值相对应的x
,y
另一方面,对于 each A
。
这样做很容易:x
(Q1)y
A.select(fn.MIN(B.x), B.z.alias('z_x')).join(B).group_by(A)
> A_id x z_x
> 0 3. 1
> 1 2. 3
我也可以为 B 做同样的事情。我的问题是:你会怎么做才能同时拥有两者?(Q2)
???
> A_id x z_x y z_y
> 0 3. 1 4. 2
> 1 2. 3 1. 4
当然,我试着做
A.select(fn.MIN(B.x), B.z.alias('z_x'), fn.MIN(B.y), B.z.alias('z_y')).join(B).group_by(A)
但该列z_x
只是替换为z_y
.
对于那里的 SQL 专家来说,(Q1)对应于
SELECT MIN("t1"."x"), "t1"."z" FROM "A" AS "t2" INNER JOIN "B" AS "t1" ON ("t1"."A_id" = "t2"."id") GROUP BY "t2"."id"
如果有人在纯 SQL 中看到解决方案,我还可以在 peewee 中翻译(Q2)的 SQL 查询!
很高兴获得任何帮助,如果需要,我可以提供更多详细信息!
解决方案
这最终比我最初想象的要难,因为 Sqlite 有点让简化版本的查询工作......但是它们并不完全正确。无论如何,以下似乎工作正常,至少就我理解的问题而言(为每个 A 获取最小 X 和最小 Y 的 Z):
class A(Base):
key = TextField()
class B(Base):
a = ForeignKeyField(A)
x = IntegerField()
y = IntegerField()
z = IntegerField()
db.create_tables([A, B])
a1 = A.create(key='a1')
a2 = A.create(key='a2')
B.create(a=a1, x=1, y=100, z=10)
B.create(a=a1, x=10, y=10, z=5)
B.create(a=a1, x=100, y=1, z=2)
B.create(a=a2, x=2, y=200, z=20)
B.create(a=a2, x=20, y=20, z=15)
B.create(a=a2, x=200, y=4, z=10)
# The idea is we will create first a query that gets us
# the minimum X for each A (subq).
BX = B.alias()
subq = BX.select(BX.a, fn.MIN(BX.x).alias('minx')).group_by(BX.a)
# Then we want to get the Z's for all rows whose X is
# equal to the minimum X (for each A).
z_for_x = (BX
.select(BX.a, BX.z, subq.c.minx)
.join(subq, on=((BX.a == subq.c.a_id) & (BX.x == subq.c.minx)))
.group_by(BX.a, BX.z))
# Do the same for the Y's.
# First find the minimum Y for each A.
BY = B.alias()
subq = BY.select(BY.a, fn.MIN(BY.y).alias('miny')).group_by(BY.a)
# Then get all Z's for each A whose Y is equal to the
# minimum Y we found.
z_for_y = (BY
.select(BY.a, BY.z, subq.c.miny)
.join(subq, on=((BY.a == subq.c.a_id) & (BY.y == subq.c.miny)))
.group_by(BY.a, BY.z))
# We'll use common table expressions to access these
# two distinct subqueries.
xcte = z_for_x.cte('zforx', columns=['ba', 'bz', 'minx'])
ycte = z_for_y.cte('zfory', columns=['ba', 'bz', 'miny'])
# Then get all A's and their associated minimum X and it's
# corresponding Z, and minimum Y and it's corresponding Z:
q = (A.select(A, xcte.c.minx, xcte.c.bz, ycte.c.miny, ycte.c.bz)
.join_from(A, xcte, JOIN.INNER, on=(A.id == xcte.c.ba))
.join_from(A, ycte, JOIN.INNER, on=(A.id == ycte.c.ba))
.with_cte(xcte, ycte))
# Execute our query:
for row in q.tuples():
print(row)
#OUTPUT:
# 1, 'a1', 1, 10, 1, 2
# 2, 'a2', 2, 20, 4, 10
输出说明:
对于 a1,最小 X=1(z=10),最小 Y=1(z=2)
对于a2,最小X=2(z=20),最小Y=4(z=10)
推荐阅读
- date - UDF 函数如何在 pyspark 中以日期为参数工作?
- r - 如何结合 unicode 在 ggplot2 中使用它?
- android - Android IAP 订阅 - 如何在接收实时开发者通知时获取原始 orderID
- javascript - 使用 useState() 方法时出错
- python - 使用python进行时间数据分析
- javascript - 我在 JavaScript 文件顶部设置的“全局”变量无法被文件内更下方的函数访问
- firebase - Firebase 云存储定价 - 什么是“下载的 GB”和“传输的 GB”
- spring-boot - 外键查找失败
- sql - 根据该 pkey 值的日期范围指定分区 #
- database - 在oracle中向表中添加列