主页 / coding / 问题 / 79266819
Accepted
Viktor
Asked: 2024-12-10 09:31:59 +0800 CST

更快的词汇表生成

  • 772

我正在尝试在 Django 中创建一个目录，queryset如下所示：

def get_toc(self):
    toc = {}
    qs = self.get_queryset()
    idx = set()
    for q in qs:
        idx.add(q.title[0])
    idx = list(idx)
    idx.sort()
    for i in idx:
        toc[i] = []
        for q in qs:
            if q.title[0] == i:
                toc[i].append(q)
    return toc

但是它有时间复杂度O(n^2)。有没有更好的方法呢？

更新 我的意思是词汇表，而不是目录。

python

3 个回答

  1. Best Answer

    这看起来不像是目录，而是词汇表，您可以在其中将术语的第一个字符映射到术语列表。

    我们可以在这里使用[python-doc]：.groupby(…) 

    from itertools import groupby

result = {
    k: list(vs)
    for k, vs in groupby(
        self.get_queryset().order_by('title'), lambda x: x.title[0]
    )
}
    • 2

  2. 我相信这相当于上面的，但应该更快，因为它使用数据库引擎进行排序和重复数据删除：

    def get_toc(self):
    return {
        q.title[0]: q
        for q in self.get_queryset().distinct("title").order_by('title')
    }
    • 1

  3. 无需遍历 中每个项目的所有查询，而是将其作为最外层且唯一的循环idx进行迭代，将每个查询添加到适当的存储桶中。如下所示：qsfor

    def get_toc(self):
    toc = {}
    qs = self.get_queryset()

    for q in qs:
        t = q.title[0]
        toc[t].setdefault(t, []).append(value)

    return toc
    • 1

相关问题