Question

我有一个清单：

nums = [1, 2, 3, 4]

我想获得分割列表1 - 3的所有可能性：

[
    ( 1, (2, 3, 4) ),
    ( 2, (1, 3, 4) ),
    ( 3, (1, 2, 4) ),
    ( 4, (1, 2 ,3) )
]

现在我能找到的最好的是使用itertools.combinations(num, 3)，但它只会给出每个项目的第二部分，这意味着[(1, 2, 3), (1, 2, 4), (1, 3, 4), (2, 3, 4)]。还有其他我可以使用的方法吗？

Answer 1

根据您解决问题的一般情况，解决方案可能或多或少简单：）

In [1]: nums = [1, 2, 3, 4]

In [2]: [(x, tuple(y for y in nums if y != x)) for x in nums]
Out[2]: [(1, (2, 3, 4)), (2, (1, 3, 4)), (3, (1, 2, 4)), (4, (1, 2, 3))]

如果列表中有重复值，请使用索引进行比较：

In [3]: [(x, tuple(y for j, y in enumerate(nums) if j != i)) for i, x in enumerate(nums)]
Out[3]: [(1, (2, 3, 4)), (2, (1, 3, 4)), (3, (1, 2, 4)), (4, (1, 2, 3))]

Answer 2

你已经到了一半。您可以使用zip和reversed将结果与原始列表合并：

result = zip(num, reversed(itertools.combinations(num, 3)))

但是如果你想使用第一部分只是某种索引，则没有必要。您可以使用列表中元素的位置作为索引（显然）。

Answer 3

这似乎是一个更简单的解决方案：

nums = [1, 2, 3, 4]
for n in range(len(nums)):
    print (nums[n], nums[:n] + nums[n+1:])

理解：

result = [(s, nums[:n] + nums[n+1:]) for n, s in enumerate(nums)]

Answer 4

这样做很有效但可能效率低下：

>>> t=set(nums) ; [ (t.difference(x).pop(),x) for x in list(itertools.combinations(t,3)) ]
[(4, (1, 2, 3)), (3, (1, 2, 4)), (2, (1, 3, 4)), (1, (2, 3, 4))]

Python列表组合

4 个答案: