我在试图实现一个算法时遇到了麻烦。我有两个列表,我想从这两个列表中获取特定的组合。
下面是一个例子。
names = ['a', 'b']
numbers = [1, 2]
在这种情况下,输出将是:
[('a', 1), ('b', 2)]
[('b', 1), ('a', 2)]
我的名字可能比数字多,比如len(names) >= len(numbers)
。下面是一个包含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)]
发布于 2015-12-02 07:57:18
最简单的方法是使用itertools.product
a = ["foo", "melon"]
b = [True, False]
c = list(itertools.product(a, b))
>> [("foo", True), ("foo", False), ("melon", True), ("melon", False)]
发布于 2016-08-21 21:33:40
可能比上面最简单的一个更简单:
>>> 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>
不需要任何导入
发布于 2019-07-24 17:07:21
找出大量列表的所有组合的最佳方法是:
import itertools
from pprint import pprint
inputdata = [
['a', 'b', 'c'],
['d'],
['e', 'f'],
]
result = list(itertools.product(*inputdata))
pprint(result)
结果将是:
[('a', 'd', 'e'),
('a', 'd', 'f'),
('b', 'd', 'e'),
('b', 'd', 'f'),
('c', 'd', 'e'),
('c', 'd', 'f')]
https://stackoverflow.com/questions/12935194
复制相似问题