来自子查询的 Django Select

Question

我想使用 window 函数进行查询，然后对子查询进行聚合分组。但我无法用 ORM 方法做到这一点。它将 return aggregate function calls cannot contain window function calls

有没有办法在不使用 .raw()

的情况下进行如下 SQL 的查询

SELECT a.col_id, AVG(a.max_count) FROM (
  SELECT col_id,
  MAX(count) OVER (PARTITION BY part_id ORDER BY part_id) AS max_count
  FROM table_one
) a
GROUP BY a.col_id;

例子

table_one

| id | col_id | part_id | count |
| -- | ------ | ------- | ----- |
| 1  | c1     | p1      | 3     |
| 2  | c2     | p1      | 2     |
| 3  | c3     | p2      | 1     |
| 4  | c2     | p2      | 4     |

首先我想在 part_id

上获得最大基数

| id | col_id | part_id | count | max_count |
| -- | ------ | ------- | ----- | --------- |
| 1  | c1     | p1      | 3     | 3         |
| 2  | c2     | p1      | 2     | 3         |
| 3  | c3     | p2      | 1     | 4         |
| 4  | c2     | p2      | 4     | 4         |

最后得到 max_count 分组的平均值 col_id

| col_id | avg(max_count) |
| ------ | -------------- |
| c1     | 3              |
| c2     | 3.5            |
| c3     | 4              |

我现在拥有的机型

def Part(models.Model):
    part_id = models.UUIDField(primary_key=True, editable=False, default=uuid.uuid4)
    name = models.CharFields()

def Col(models.Model):
    part_id = models.UUIDField(primary_key=True, editable=False, default=uuid.uuid4)
    name = models.CharFields()

def TableOne(models.Model):
    id = models.UUIDField(primary_key=True, editable=False, default=uuid.uuid4)
    col_id = models.ForeignKey(
        Col,
        on_delete=models.CASCADE,
        related_name='table_one_col'
    )
    part_id = models.ForeignKey(
        Part,
        on_delete=models.CASCADE,
        related_name='table_one_part'
    )
    count = models.IntegerField()

我想在partition by之后做group by。这是我做的会带来错误的查询。

query = TableOne.objects.annotate(
    max_count=Window(
        expression=Max('count'),
        order_by=F('part_id').asc(),
        partition_by=F('part_id')
    )
).values(
    'col_id'
).annotate(
    avg=Avg('max_count')
)

Answer 1

在Django中可以使用subqueries，不需要使用window函数。首先，子查询是一个 Part 查询集，用 TableOne

中的最大计数进行注释

from django.db.models import Avg, Max, Subquery, OuterRef

parts = Part.objects.filter(
    id=OuterRef('part_id')
).annotate(
    max=Max('table_one_part__count')
)

然后用子查询的最大计数注释一个 TableOne 查询集，对我们要分组的列执行 values (col_id ) 然后用平均值再次注释生成您想要的输出

TableOne.objects.annotate(
    max_count=Subquery(parts.values('max')[:1])
).values(
    'col_id'
).annotate(
    Avg('max_count')
)

来自子查询的 Django Select

Django Select from a Subquery

python

django

django-models

python-3.x

django-rest-framework