python中的乘法字符串向量

Question

我有动态数量的字符串向量，我需要将它们相乘，并创建一个dict数组，将所有可能的结果结合起来

如果输入是静态的，我会这样做：

sth = {
    "param1": ["p1v1", "p1v2"],
    "param2": ["p2v1", "p2v2", "p2v3"],
    "param3": ["p3v1", "p3v2"]
}

param_sets = []

for a_top_val in sth["param1"]:
    for a_val in sth["param2"]:
        for a_bottom_val in sth["param3"]:
            param_sets.append({
                "param1": a_top_val,
                "param2": a_val,
                "param3": a_bottom_val
            })

print(param_sets)
print(len(param_sets))

这将打印组合的dict，如下所示：

[
    {'param1': 'p1v1', 'param2': 'p2v1', 'param3': 'p3v1'},
    {'param1': 'p1v1', 'param2': 'p2v1', 'param3': 'p3v2'},
    {'param1': 'p1v1', 'param2': 'p2v2', 'param3': 'p3v1'},
    {'param1': 'p1v1', 'param2': 'p2v2', 'param3': 'p3v2'},
    {'param1': 'p1v1', 'param2': 'p2v3', 'param3': 'p3v1'},
    {'param1': 'p1v1', 'param2': 'p2v3', 'param3': 'p3v2'},
    {'param1': 'p1v2', 'param2': 'p2v1', 'param3': 'p3v1'},
    {'param1': 'p1v2', 'param2': 'p2v1', 'param3': 'p3v2'},
    {'param1': 'p1v2', 'param2': 'p2v2', 'param3': 'p3v1'},
    {'param1': 'p1v2', 'param2': 'p2v2', 'param3': 'p3v2'},
    {'param1': 'p1v2', 'param2': 'p2v3', 'param3': 'p3v1'},
    {'param1': 'p1v2', 'param2': 'p2v3', 'param3': 'p3v2'}
]

如何使sth周围的for循环成为动态的

Answer 1

您可以将sth中的每个条目转换为键值元组列表。然后可以使用itertools.product获取所有列表的笛卡尔乘积，并使用该乘积创建参数字典：

import itertools as it

sth = {
    "param1": ["p1v1", "p1v2"],
    "param2": ["p2v1", "p2v2", "p2v3"],
    "param3": ["p3v1", "p3v2"]
}

param_tuples = (
    [(param, val) for val in values] for param, values in sth.items()
)

result = [
    dict(param_combination) for param_combination in it.product(*param_tuples)
]

结果:

[{'param1': 'p1v1', 'param2': 'p2v1', 'param3': 'p3v1'},
 {'param1': 'p1v1', 'param2': 'p2v1', 'param3': 'p3v2'},
 {'param1': 'p1v1', 'param2': 'p2v2', 'param3': 'p3v1'},
 {'param1': 'p1v1', 'param2': 'p2v2', 'param3': 'p3v2'},
 {'param1': 'p1v1', 'param2': 'p2v3', 'param3': 'p3v1'},
 {'param1': 'p1v1', 'param2': 'p2v3', 'param3': 'p3v2'},
 {'param1': 'p1v2', 'param2': 'p2v1', 'param3': 'p3v1'},
 {'param1': 'p1v2', 'param2': 'p2v1', 'param3': 'p3v2'},
 {'param1': 'p1v2', 'param2': 'p2v2', 'param3': 'p3v1'},
 {'param1': 'p1v2', 'param2': 'p2v2', 'param3': 'p3v2'},
 {'param1': 'p1v2', 'param2': 'p2v3', 'param3': 'p3v1'},
 {'param1': 'p1v2', 'param2': 'p2v3', 'param3': 'p3v2'}]

Answer 2

使用itertools.product的组合获取所有可能值的组合，使用zip()将键重新应用于元组：

from itertools import product

sth = {
    "param1": ["p1v1", "p1v2"],
    "param2": ["p2v1", "p2v2", "p2v3"],
    "param3": ["p3v1", "p3v2"]
}

prod = product(*(sth[k] for k in sth))
a = [dict(zip(sth.keys(), p)) for p in prod]

print(a)

要获得：

[{'param1': 'p1v1', 'param2': 'p2v1', 'param3': 'p3v1'},
{'param1': 'p1v1', 'param2': 'p2v1', 'param3': 'p3v2'},
{'param1': 'p1v1', 'param2': 'p2v2', 'param3': 'p3v1'},
{'param1': 'p1v1', 'param2': 'p2v2', 'param3': 'p3v2'},
{'param1': 'p1v1', 'param2': 'p2v3', 'param3': 'p3v1'},
{'param1': 'p1v1', 'param2': 'p2v3', 'param3': 'p3v2'},
{'param1': 'p1v2', 'param2': 'p2v1', 'param3': 'p3v1'},
{'param1': 'p1v2', 'param2': 'p2v1', 'param3': 'p3v2'},
{'param1': 'p1v2', 'param2': 'p2v2', 'param3': 'p3v1'},
{'param1': 'p1v2', 'param2': 'p2v2', 'param3': 'p3v2'},
{'param1': 'p1v2', 'param2': 'p2v3', 'param3': 'p3v1'},
{'param1': 'p1v2', 'param2': 'p2v3', 'param3': 'p3v2'}]

Answer 3

要在python中循环字典，可以使用

for k, v in sth:
    print (k, v)