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"}
]
Seth Johnson asked 2020-01-24T02:43:46Z
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
translate from https://stackoverflow.com:/questions/5228158/cartesian-product-of-a-dictionary-of-lists