如何从列表的列表中生成索引不重复的所有可能组合？

你的需求是生成所有元素组合，要求组合中每个元素的子列表索引互不重复（即不能有两个元素来自不同/相同子列表的同一位置），同时支持从单个子列表中选取多个不同索引的元素。

核心思路 #

1. 按索引分组：把所有子列表中同一索引位置的元素归为一组（比如索引0的组包含所有子列表的第0个元素，索引1的组包含所有子列表的第1个元素）。
2. 生成索引子集：选取所有可能的不重复索引组合（比如选[0,1]、[0,1,2]等）。
3. 生成元素组合：对每个索引子集，从对应的元素组中各选一个元素，生成所有可能的笛卡尔积，这些就是符合要求的组合。

Python 实现代码 #

from itertools import combinations, product

# 示例输入（替换成你的实际列表）
input_lists = [
    ['a1', 'a2', 'a3'],
    ['b1', 'b2', 'b3'],
    ['c1', 'c2', 'c3']
]

# 按索引分组：key为索引，value为所有子列表中该索引的元素
n = len(input_lists[0])  # 假设所有子列表长度一致
index_to_elements = {idx: [lst[idx] for lst in input_lists] for idx in range(n)}

all_valid_combinations = []

# 遍历所有可能的组合长度（从1到n）
for combo_length in range(1, n + 1):
    # 生成所有长度为combo_length的不重复索引组合
    for index_group in combinations(range(n), combo_length):
        # 获取每个索引对应的元素组
        element_groups = [index_to_elements[idx] for idx in index_group]
        # 生成该索引组下的所有元素组合（笛卡尔积）
        for combo in product(*element_groups):
            all_valid_combinations.append(list(combo))

# 打印结果示例（可按需注释或删除）
for combo in all_valid_combinations[:10]:
    print(combo)

代码说明 #

• combinations(range(n), combo_length)：生成所有长度为combo_length的不重复索引组合，从根源保证组合内的索引互不相同。
• product(*element_groups)：对每个索引对应的元素组生成笛卡尔积，既支持跨子列表选不同索引的元素（比如['a1','b2','c3']），也支持从同一子列表选多个不同索引的元素（比如['a1','a2']）。
• 最终all_valid_combinations包含了所有符合要求的组合，覆盖了你举例的所有有效场景。

内容的提问来源于stack exchange，提问作者Anatol Melaku