如何在Python中遍历YAML文件生成所有列表元素的可能组合

嘿，你已经搞定了YAML读取的部分，接下来要实现从每个列表各取一个元素、生成所有可能组合且覆盖所有列表的需求其实很简单——Python标准库中的itertools.product就是专门干这个的，它能计算多个可迭代对象的笛卡尔积，完美匹配你的场景。

完整可运行代码 #

#!/usr/bin/env python
import yaml
from itertools import product

# 安全读取YAML文件（推荐用safe_load替代load，避免恶意内容风险）
with open("parameters.yaml", 'r') as stream:
    try:
        yaml_data = yaml.safe_load(stream)
    except yaml.YAMLError as exc:
        print(f"读取YAML出错: {exc}")
        exit()

# 提取所有列表（按YAML中的键顺序）
all_lists = list(yaml_data.values())

# 生成所有组合：每个列表取一个元素的全排列
combinations = product(*all_lists)

# 打印所有组合（也可根据业务需求进一步处理）
print("所有可能的组合：")
for idx, combo in enumerate(combinations, 1):
    print(f"{idx}. {combo}")

# 可选：生成带原键的字典形式组合（更直观）
print("\n带键的字典组合：")
keys = list(yaml_data.keys())
dict_combinations = [dict(zip(keys, combo)) for combo in product(*all_lists)]
for idx, combo_dict in enumerate(dict_combinations, 1):
    print(f"{idx}. {combo_dict}")

关键部分说明 #

1. 安全读取YAML：替换yaml.load为yaml.safe_load是官方推荐的做法，能避免加载恶意构造的YAML内容带来的安全隐患。
2. 提取列表集合：通过yaml_data.values()拿到YAML中所有的列表，转成列表后传给product即可。
3. 笛卡尔积生成：product(*all_lists)里的*是解包操作，把每个列表作为独立参数传入，这样product会自动从每个列表取一个元素，生成所有可能的组合，保证每个列表都被用到。
4. 字典格式优化：如果需要明确知道每个元素来自哪个YAML键，可以用zip(keys, combo)生成字典，结果会更直观易读。

示例输出 #

基于你提供的YAML内容，运行代码后会输出：

所有可能的组合：
1. ('a', 'd', 'f')
2. ('a', 'd', 'g')
3. ('a', 'd', 'h')
4. ('a', 'd', 'i')
5. ('a', 'e', 'f')
...（后续所有组合）

带键的字典组合：
1. {'One': 'a', 'Two': 'd', 'Three': 'f'}
2. {'One': 'a', 'Two': 'd', 'Three': 'g'}
...（后续所有组合）

这个方案用的都是Python标准库，不需要额外安装依赖，简洁又高效，完全满足你的需求~

内容的提问来源于stack exchange，提问作者Bhavana Mehta