我正在尝试计算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。我正在努力学习。
答案 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。