如果在Django ORM中没有特定年份的值,如何使用groupby获取列表和分组中的所有值,并将0放入列表中?

2024-04-27 01:07:33 发布

您现在位置:Python中文网/ 问答频道 /正文

我有两个模型。一个是带有“分行id”和“分行名称”字段的分行,另一个是带有“销售额”和“年份”字段的销售。分公司和销售部之间有着外键关系。一个分支可以有多个销售。我所要达到的目标是按年度分组,其中每年都有其销售价值。并获取键(分支名称)和值(sales\u list)的字典如果该年没有销售,那么应该有值“0”。你知道吗

你知道吗型号.py你知道吗

import datetime
from django.db import models


class Branch(models.Model):

    branch_id = models.IntegerField()
    name = models.CharField(max_length=50)


    def __str__(self):
        return self.name

class Sales(models.Model):
    branch = models.ForeignKey('Branch', on_delete = models.CASCADE)
    sales_amount = models.IntegerField()
    year = models.CharField(max_length=4) 

    class Meta:
        unique_together = (('year', 'branch'),)

    def __str__(self):
        return self.branch.name

你知道吗视图.py你知道吗

from django.shortcuts import render
import json


from .models import Branch, Sales


def index(request):
    branches = Branch.objects.all()
    sales = Sales.objects.all()

    sales_list = []
    branch_list = []
    year_set = set()

    for sale in sales:
        year_set.add(sale.year)

    year_list = list(sorted(year_set))

    for branch in branches:
        sales_list.append(branch.sales_set.all().order_by('year'))
        branch_list.append(branch.name)

    sales_list = [[j.sales_amount for j in i] for i in sales_list]

    branch_dict = dict(zip(branch_list, sales_list))

    return render(request, 'index.html', {
        'year_list': year_list,
        'branch_dict': branch_dict
    })


我已经

BRANCH001 with the year 2011 has 5000 sales, BRANCH001 with the year 2012 has 10000 salesBRANCH002 with the year 2011 has 7000 sales

expected output = {'BRANCH001': [5000, 10000], 'BRANCH002': [0, 7000]}

actual output = {'BRANCH001': [5000, 10000], 'BRANCH002': [7000]}


Tags: nameinfromimportselfbranchformodels