python - 在 Django 中生成给定查询集的所有可能子集

Question

这只是一个示例，但给出了以下模型:

class Foo(models.model):
    bar = models.IntegerField()

    def __str__(self):
        return  str(self.bar)

    def __unicode__(self):
        return str(self.bar)

以及以下QuerySet对象:

foobar = Foo.objects.filter(bar__lt=20).distinct()

(意思是，一组独特的 Foo 模型与 bar <= 20 )，我如何生成 foobar 的所有可能子集？ ？理想情况下，我想进一步限制子集，以便对于每个子集 x的 foobar , 所有 f.bar 的总和在 x (其中 f 是类型为 Foo 的模型)介于某个最大值和最小值之间。

因此，例如，给定以下 foobar 的实例:

>> print foobar
[<Foo: 5>, <Foo: 10>, <Foo: 15>]

和min=5 , max=25 ，我想构建一个如下所示的对象(最好是 QuerySet，但可能是一个列表):

[[<Foo: 5>], [<Foo: 10>], [<Foo: 15>], [<Foo: 5>, <Foo: 10>],
 [<Foo: 5>, <Foo: 15>], [<Foo: 10>, <Foo: 15>]]

我试验过 itertools但它似乎并不是特别适合我的需要。

我认为这可以用复杂的 QuerySet 来完成但我不确定如何开始。

Answer 1

S = [list(itertools.combinations(foobar,i)) for i in xrange(1, len(foobar))]

它产生非平面列表。您可以通过以下方式将其展平:

list(itertools.chain.from_iterable(S))