如何基于生成具有给定范围内的元素的给定大小的列表来创建列表理解？ [关闭]

Question

给定两个参数：

• dim子列表的长度
• n 这个列表中每个元素的取值范围

我想生成一个列表列表，以便结果包含子列表的所有唯一组合（对于dim 和n）

例如：

dim = 3
n = 2

我要生成：

[
    [0, 0, 0],
    [0, 0, 1],
    [0, 1, 0],
    [0, 1, 1],
    [1, 0, 0],
    [1, 0, 1],
    [1, 1, 0],
    [1, 1, 1]
]

为：

dim = 2
n = 3

我要生成：

[
    [0, 0],
    [0, 1],
    [0, 2],
    [1, 0],
    [1, 1],
    [1, 2],
    [2, 0],
    [2, 1],
    [2, 2]
]

我试过了：

[[a,b] for a in range(dim) for b in range(n)]

但这并不完全正确。一段时间以来，我一直在为此挠头。我只是不知道该怎么做。

注意，我假设列表推导是要走的路。也欢迎其他方法。但我认为列表推导是一个好方法，因为我稍后会添加更多限制。喜欢

[[a,b] for a in range(dim) for b in range(n) if a!=b]

最后一个条件a!=b 是一个非常简单的条件。实际上，它们会更多地出现在“如果a 属于这个集合，而b 属于另一个集合并且ab+4`”的形式上。

此代码的目标是创建一个函数，该函数可以接收像吉他这样的按弦乐器，接收弦数、它们的调音和和弦，然后它应该吐出可能的指法方式。但是，我担心性能，因为我想处理至少六个带有十二个不同音符的弦。但这是我以后会担心的问题。

所以完整的样本可能是这样的。 tuning 只是一个列表，其中大小是字符串的数量，元素是调整。而chord 是某种定义有效输出的结构。

>>> get_fingerings(tuning, chord)
[[6,6,4,3,0,0], [1,3,0,0,3]]

Answer 1

您可以将列表推导与itertools.product 一起用作元组的生成器，以及在最终列表中包含或排除它们的任何过滤器。例如：

[item for item 
    in itertools.product(range(5), range(5)) # generate
    if item[0] != item[1]]                   # filter in/out

Answer 2

如果条件只是a != b != ...，那么只需简单地使用itertools.permutations：

[i for i in permutations(range(n), dim)]

喜欢： [i for i in permutations(range(2), 2)] => [(0, 1), (1, 0)]