python-列表字典的笛卡尔积

我正在尝试编写一些代码来测试一堆输入参数的笛卡尔积。

我看过itertools，但是它的product功能并不是我想要的。 是否有一种简单明了的简单方法来制作一个字典，每个字典中有任意数量的键和任意数量的元素，然后生成具有下一个排列的字典？

输入：

options = {"number": [1,2,3], "color": ["orange","blue"] }

print list( my_product(options) )

输出示例：

[ {"number": 1, "color": "orange"},

{"number": 1, "color": "blue"},

{"number": 2, "color": "orange"},

{"number": 2, "color": "blue"},

{"number": 3, "color": "orange"},

{"number": 3, "color": "blue"}

]

4个解决方案

49 votes

好的，谢谢@dfan告诉我我在错误的地方看。 我现在知道了：

from itertools import product

def my_product(inp):

return (dict(zip(inp.keys(), values)) for values in product(*inp.values())

编辑：经过多年的Python经验，我认为一个更好的解决方案是接受list(product_dict(**mydict))，而不是输入字典。 调用样式更类似于原始的kwargs。我还认为编写生成器函数(而不是返回生成器表达式的函数)可使代码更清晰。 所以：

def product_dict(**kwargs):

keys = kwargs.keys()

vals = kwargs.values()

for instance in itertools.product(*vals):

yield dict(zip(keys, instance))

如果需要传递字典list(product_dict(**mydict))。使用kwargs(而不是任意输入类)的一个显着变化是，它防止对键/值进行排序，至少直到Python 3.6为止。

Seth Johnson answered 2020-01-24T02:44:36Z

17 votes

塞思答案的Python 3版本。

import itertools

def dict_product(dicts):

"""

>>> list(dict_product(dict(number=[1,2], character='ab')))

[{'character': 'a', 'number': 1},

{'character': 'a', 'number': 2},

{'character': 'b', 'number': 1},

{'character': 'b', 'number': 2}]

"""

return (dict(zip(dicts, x)) for x in itertools.product(*dicts.values()))

Tarrasch answered 2020-01-24T02:44:56Z

7 votes

顺便说一下，这不是排列。 排列是列表的重排。 这是从列表中可能选择的列举。

编辑：在记住它被称为笛卡尔积之后，我想到了这个：

import itertools

options = {"number": [1,2,3], "color": ["orange","blue"] }

product = [x for x in apply(itertools.product, options.values())]

print [dict(zip(options.keys(), p)) for p in product]

dfan answered 2020-01-24T02:45:20Z

3 votes

# I would like to do

keys,values = options.keys(), options.values()

# but I am not sure that the keys and values would always

# be returned in the same relative order. Comments?

keys = []

values = []

for k,v in options.iteritems():

keys.append(k)

values.append(v)

import itertools

opts = [dict(zip(keys,items)) for items in itertools.product(*values)]

结果是

opts = [

{'color': 'orange', 'number': 1},

{'color': 'orange', 'number': 2},

{'color': 'orange', 'number': 3},

{'color': 'blue', 'number': 1},

{'color': 'blue', 'number': 2},

{'color': 'blue', 'number': 3}

]

Hugh Bothwell answered 2020-01-24T02:45:40Z