在python中乘以字符串向量

Question

我有动态数量的字符串向量，我需要将它们相乘并创建一个组合所有可能性的字典数组。

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

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))

这会打印这样的组合字典：

[
    {'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

使用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 2

您可以将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 3

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

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