Question

例如：

list1 = [5,8]
list2 = [4,4,2,3,6]

使用powerset函数很容易在list2中获得5和8的组合

def powerset(iterable):
    "powerset([1,2,3]) --> () (1,) (2,) (3,) (1,2) (1,3) (2,3) (1,2,3)"
    s = list(iterable)
    return chain.from_iterable(combinations(s, r) for r in range(len(s)+1))

8可以由[4,4]或[2,6]组成，但5只能由[2,3]组成。如果我选择[2,6]代表8，则list2中没有5个组合。

如何获得8的[4,4]和5的[2,3]？我想在list2中为list1中的数字选择尽可能多的组合。实际上list1中的数字可能包含list2中的3个或更多数字。

实际问题更加困难，因为list1中可能没有使用某些数字，list1中的数字可能包含3个或更多数字。

Answer 1

我认为这可以满足您的需求：

list1 = [5,8]
list2 = [4,4,2,3,6]


for i in list2:
    for j in list2:
        if i+j in list1:
            print("%d + %d = %d"%(i, j, i+j))

它尝试创建所有可能的添加项，如果它包含在第一个列表中，则输出它。

输出：

4 + 4 = 8
4 + 4 = 8
4 + 4 = 8
4 + 4 = 8
2 + 3 = 5
2 + 6 = 8
3 + 2 = 5
6 + 2 = 8

Answer 2

这是一种简洁有效的方法。

import itertools

def combos(a, b):

    matches = []
    for combo in list(itertools.combinations(a, 2)):
        if combo[0] + combo[1] in b:
            matches.append(combo)
return matches    

>> [(4, 4), (2, 3), (2, 6)]

这是另一种方式：

def foo(matches, *args):

    matches_dict = {k: [] for k in matches}
    for _, tup in enumerate(*args):
        if sum(tup) in matches:
            matches_dict[sum(tup)].append(tup)
    return matches_dict

>> {5: [(2, 3)], 8: [(4, 4), (2, 6)]}

现在计时：

time combos(list2, list1)
CPU times: user 23 µs, sys: 7 µs, total: 30 µs
Wall time: 31 µs

time foo(list1, list(itertools.combinations(list2, 2)))
CPU times: user 33 µs, sys: 9 µs, total: 42 µs
Wall time: 40.1 µs

使用@moritzg答案，修改为不包括dupes，

def fizz(list1, list2):
    matches = []
    for i in list2:
        for j in list2:
            if i+j in list1:
                matches.append((i,j))
    return set(matches)

time fizz(list1, list2)
CPU times: user 26 µs, sys: 13 µs, total: 39 µs
Wall time: 35 µs

此外，我忘了提及(2,6)与(6,2)的区别是否与itertools.combinations不同，但您可以将itertools.permutations切换为Unable to import module 'lambda_function': No module named pandas。

如何找到尽可能多的列表完整分区？

2 个答案: