Django加权百分比

时间:2017-09-07 16:58:30

标签: python django python-3.x

我正在尝试计算Django查询中的加权百分比。

这是我的数据的示例:

id      start_date  agency_id   area_id housetype_id    no_of_changed   price_change_percentage total
6716    2017-08-26  11          1       1               16              -0.09                   35
6717    2017-08-26  11          1       3               44              -0.11                   73
6718    2017-08-26  11          1       4               7               -0.1                    12
6719    2017-08-26  11          1       5               0               0                       4
6720    2017-08-26  11          1       6               0               0                       1
6721    2017-08-26  21          1       1               0               0                       1
6722    2017-08-26  34          1       1               0               0                       1
6723    2017-08-26  35          1       1               0               0                       1
6724    2017-08-26  38          1       1               0               0                       1

这是我目前的代码:

from django.db.models import F, FloatField, ExpressionWrapper
from app.models import PriceChange

def weighted_percentage(area_id, date_range, agency_id, housetype):

    data = PriceChange.objects.filter(area_id=area_id,
                                      start_date__range=date_range,
                                      agency_id=agency_id,
                                      )

    if housetype:
        data = data.filter(housetype=housetype) \
            .values('start_date') \
            .annotate(price_change_total=ExpressionWrapper((F('price_change_percentage') * F('no_of_changed')) / F('total'), output_field=FloatField())) \
            .order_by('start_date')

    else:
        # what to do?
        pass

    x = [x['start_date'] for x in data]
    y = [y['price_change_total'] for y in data]

    return x, y

我想出了在定义housetype时如何进行计算,我只需要从一行中获取数据。当我需要使用相同的start_date计算多行时,我无法弄清楚如何完成。我不希望每行都有一个值,而是每个start_date。

作为示例(两行具有相同的start_date,area_id,agency_id但不同的housetype_ids):     no_of_changed price_change_percentage总计     16 -0.09 35     44 -0.11 73

计算是伪代码: ((no_of_changed[0] * price_changed_percentage[0]) + (no_of_changed[0] * price_changed_percentage[0])) / (total[0] + total[1]) = price_change_total

((16 * -0.09) + (44 * -0.11)/ (35 + 73) = -0.03148148

我正在使用Django 1.11和Python 3.6。如果问题不清楚或需要更多信息或其他任何错误请告诉我(所以我可以调整),不要只是downvote。我正在努力学习。

1 个答案:

答案 0 :(得分:1)

您需要将表达式包装在Sum表达式中。

添加以下导入:

from django.db.models import Sum

然后添加以下查询

else:
    data = data.values('start_date') \
       .annotate(
           price_change_total=ExpressionWrapper(
               Sum(F('price_change_percentage') * F('no_of_changed')) / Sum(F('total')),
               output_field=FloatField()
           )
       ) \
       .order_by('start_date')

这里发生的事情是,当您在Sum调用中使用annotate()之类的聚合表达式时,会将其转换为数据库中的group_by查询。上一个values()子句中列出的所有列都用于创建group_by查询。

有关进一步说明和生成的SQL查询的详细信息,请参阅this blog post