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