我很难理解我要实现的一个算法。我有两个链表，我想从这两个链表中取特定的组合。
举个例子。
names = ['a', 'b']
numbers = [1, 2]
这种情况下的输出将是:
[('a', 1), ('b', 2)]
[('b', 1), ('a', 2)]
我可能有更多的名称比数字，即len(名称)>= len(数字)。下面是一个有3个名字和2个数字的例子:
names = ['a', 'b', 'c']
numbers = [1, 2]
输出:
[('a', 1), ('b', 2)]
[('b', 1), ('a', 2)]
[('a', 1), ('c', 2)]
[('c', 1), ('a', 2)]
[('b', 1), ('c', 2)]
[('c', 1), ('b', 2)]
当前回答
对interjay的答案进行了微小的改进，使结果作为一个平坦的列表。
>>> list3 = [zip(x,list2) for x in itertools.permutations(list1,len(list2))]
>>> import itertools
>>> chain = itertools.chain(*list3)
>>> list4 = list(chain)
[('a', 1), ('b', 2), ('a', 1), ('c', 2), ('b', 1), ('a', 2), ('b', 1), ('c', 2), ('c', 1), ('a', 2), ('c', 1), ('b', 2)]
来自此链接的引用
其他回答
最简单的方法是使用itertools.product:
a = ["foo", "melon"]
b = [True, False]
c = list(itertools.product(a, b))
>> [("foo", True), ("foo", False), ("melon", True), ("melon", False)]
更好的答案只适用于所提供的特定长度的列表。
这里有一个版本，适用于任何长度的输入。同时，从组合和排列的数学概念上明确了算法。
from itertools import combinations, permutations
list1 = ['1', '2']
list2 = ['A', 'B', 'C']
num_elements = min(len(list1), len(list2))
list1_combs = list(combinations(list1, num_elements))
list2_perms = list(permutations(list2, num_elements))
result = [
tuple(zip(perm, comb))
for comb in list1_combs
for perm in list2_perms
]
for idx, ((l11, l12), (l21, l22)) in enumerate(result):
print(f'{idx}: {l11}{l12} {l21}{l22}')
这个输出:
0: A1 B2
1: A1 C2
2: B1 A2
3: B1 C2
4: C1 A2
5: C1 B2
你可能想尝试一行列表理解:
>>> [name+number for name in 'ab' for number in '12']
['a1', 'a2', 'b1', 'b2']
>>> [name+number for name in 'abc' for number in '12']
['a1', 'a2', 'b1', 'b2', 'c1', 'c2']
可能比上面最简单的一个更简单:
>>> a = ["foo", "bar"]
>>> b = [1, 2, 3]
>>> [(x,y) for x in a for y in b] # for a list
[('foo', 1), ('foo', 2), ('foo', 3), ('bar', 1), ('bar', 2), ('bar', 3)]
>>> ((x,y) for x in a for y in b) # for a generator if you worry about memory or time complexity.
<generator object <genexpr> at 0x1048de850>
没有任何进口
或者用KISS回答简短的清单:
[(i, j) for i in list1 for j in list2]
性能不如itertools，但你使用的是python，所以性能已经不是你最关心的…
我也喜欢其他的答案!
