在python中组合列表

时间:2014-07-20 17:20:31

标签: python list python-2.7 combinations

我正在尝试合并2个列表并希望组合。

a = ['ibm','dell']
b = ['strength','weekness']

我希望形成像['ibm strength','ibm weekness','dell strength','dell weakness']这样的组合。

我尝试使用zip或连接列表。我也使用了itertools,但它没有给我想要的输出。请帮忙。

a = ['ibm','dell']
b = ['strength','weekness']
c = a + b
itertools.combinations(c,2)
for a in a:
    for b in b:
        print a +b

2 个答案:

答案 0 :(得分:6)

您正在寻找product()。试试这个:

import itertools

a = ['ibm', 'dell']
b = ['strength', 'weakness']

[' '.join(x) for x in itertools.product(a, b)]
=> ['ibm strength', 'ibm weakness', 'dell strength', 'dell weakness']

要遍历结果,请不要忘记itertools.product()返回只能使用一次的迭代器。如果您以后需要它,请将其转换为列表(如上所述,使用列表推导)并将结果存储在变量中,以备将来使用。例如:

lst = list(itertools.product(a, b))
for a, b in lst:
    print a, b

答案 1 :(得分:0)

对于Cartesian product,您需要itertools.product()而不是组合。

嵌套的for循环也可以工作:

for x in a:
    for y in b:
        c = a + b
        print(c)