Python列表推导式 - 需要跳过元素的组合

Question

对于这个输入列表

[0, 1, 2, 3, 4, 5]

我需要这个输出结果。

[[0, 2],
 [0, 3],
 [0, 4],
 [0, 5],
 [1, 3],
 [1, 4],
 [1, 5],
 [2, 4],
 [2, 5],
 [3, 5],
 [0, 2, 3],
 [0, 3, 4],
 [0, 4, 5],
 [1, 3, 4],
 [1, 4, 5],
 [2, 4, 5],
 [0, 2, 3, 4],
 [0, 3, 4, 5],
 [1, 3, 4, 5]]

我尝试过这段代码，

for k in range( 0, 5 ):
    for i in range( len( inputlist ) - ( 2 + k ) ):
        print [inputlist[k], inputlist[i + ( 2 + k )]]
    for i in range( len( inputlist ) - ( 3 + k ) ):
        print [inputlist[k], inputlist[i + ( 2 + k )], inputlist[i + ( 3 + k )]]
    for i in range( len( inputlist ) - ( 4 + k ) ):
        print [inputlist[k], inputlist[i + ( 2 + k )], inputlist[i + ( 3 + k )], inputlist[i + ( 4 + k )]]

我需要跳过一些模式：

1,2,3 --> 1,3

1,2,3,4 --> [1,3],[1,4],[2,4]

也就是说，跳过第一个元素、第三个元素等等。

如何推广这种方法？感谢您的帮助。

- Rajasankar

---

对于1、2、3、4，您不需要[1、3、4]吗？ - Anand S Kumar

Answer 1

尝试用文字描述您的问题。

从您的示例中我所理解的是：

def good(x): return x[0]+1!=x[1] and all(i+1==j for i,j in zip(x[1:],x[2:]))

from itertools import combinations
[i for j in range(2,5) for i in filter(good, combinations(l,j))]

[(0, 2), (0, 3), (0, 4), (0, 5), (1, 3), (1, 4), (1, 5), (2, 4), (2, 5), (3, 5), (0, 2, 3), (0, 3, 4), (0, 4, 5), (1, 3, 4), (1, 4, 5), (2, 4, 5), (0, 2, 3, 4), (0, 3, 4, 5), (1, 3, 4, 5)]。这是一个包含多个元组的列表，其中每个元组都由数字组成。

- fferri

---

谢谢@mescalinum，这正是我需要的。非常感谢你的帮助。 - Rajasankar

不客气！我稍微修改了我的答案，更好地将“好”组合的条件与生成长度为2到5的所有组合分开。我认为这样读起来更好。 - fferri

Answer 2

from itertools import combinations

a=range(6)
combs=[list(combinations(a,j)) for j in range(2,5)]

combs1=[list(i) for elem in combs for i in elem if (len(i)==2 and i[1] -i[0] >1)  or (len(i)>=3 and i[-1]-i[-2] ==1 and i[1] -i[0]>1 and i[2]-i[1] ==1)]

print combs1
#Output
[[0, 2], [0, 3], [0, 4], [0, 5], [1, 3], [1, 4], [1, 5], [2, 4], [2, 5], [3, 5], [0, 2, 3], [0, 3, 4], [0, 4, 5], [1, 3, 4], [1, 4, 5], [2, 4, 5], [0, 2, 3, 4], [0, 3, 4, 5], [1, 3, 4, 5]]

- Ajay