如何在Python中生成给定元素集的所有排列组合并计算概率？

我现有如下Python代码：

mainList = ['a', 'b', 'c', 'd']
itemList = []
for i in range(0, len(mainList)):
    strs = ""
    for j in range(0, i+1):
        strs += mainList[j]
    itemList.append(strs)

inp = ""
for y in mainList:
    inp += y

for i in range(0, len(inp)):
    prob = ""
    for j in range(0, i+1):
        prob += inp[j]
    if(itemList.__contains__(prob)):
        if len(prob) > 1:
            print("Probability of " + prob + " is 100%")
        else:
            print("Probability of " + prob + " is " + str(round((1/len(mainList))*100, 0)) + "%")
    else:
        print("Probability of " + prob + " is 0%")

该代码当前输出：

Probability of a is 25.0%
Probability of ab is 100%
Probability of abc is 100%
Probability of abcd is 100%

我希望修改代码，使其能生成给定元素集（此处为4个元素）的所有排列组合（如bc、cc、dd、aa、bb、bca、aaa等所有可能序列），并计算对应的概率。我尝试自行修改但不知从何入手，恳请帮助。

要生成所有包含重复元素的任意长度序列，并按原逻辑计算概率，可借助itertools.product实现：

import itertools

main_list = ['a', 'b', 'c', 'd']
# 生成原列表的所有前缀子序列
prefixes = []
for i in range(1, len(main_list) + 1):
    prefixes.append(''.join(main_list[:i]))

# 设置要生成的序列最大长度，可按需调整
max_sequence_length = 3

# 遍历所有可能的序列长度（从1到设定的最大长度）
for length in range(1, max_sequence_length + 1):
    # 生成该长度下的所有可能序列（允许元素重复）
    for seq_tuple in itertools.product(main_list, repeat=length):
        sequence = ''.join(seq_tuple)
        # 按规则计算概率并输出
        if sequence in prefixes:
            if len(sequence) == 1:
                probability = round((1 / len(main_list)) * 100, 0)
                print(f"Probability of {sequence} is {probability}%")
            else:
                print(f"Probability of {sequence} is 100%")
        else:
            print(f"Probability of {sequence} is 0%")

关键说明： #

• 生成前缀列表：和原代码逻辑一致，提前生成main_list的所有前缀（如a、ab、abc、abcd）。
• 生成全量序列：itertools.product会生成指定长度下，元素可重复的所有组合，覆盖你需要的aa、bc、bca等所有情况。
• 概率判断：完全沿用原代码的概率规则——前缀序列中，单个元素概率为(1/元素总数)*100%，长于1的前缀概率100%；非前缀序列概率0%。

你可以修改max_sequence_length的值，来生成更长的目标序列。

内容的提问来源于stack exchange，提问作者Paul Adeleke