Python枚举三个骰子点数和为5的事件及代码问题修复

原代码的问题分析 #

• 组合生成逻辑错误：原代码通过手动重复列表元素的方式构造red、blue、green列表，无法生成三个骰子的全部笛卡尔积组合，导致最终筛选出的符合条件的结果不完整。
• 莫名数字2的来源：原代码中blue和green的列表生成语句存在换行，可能触发语法解析时的额外输出，但核心问题仍在于组合生成的逻辑错误。

修正后的代码 #

import itertools

# 定义骰子的面数
dice_faces = [1, 2, 3, 4, 5, 6]
# 生成三个骰子的所有可能组合（笛卡尔积）
all_rolls = list(itertools.product(dice_faces, repeat=3))

# 筛选出点数和为5的组合
event_a = [roll for roll in all_rolls if sum(roll) == 5]

# 格式化输出结果
print("点数和为5的所有骰子组合：")
for idx, roll in enumerate(event_a, 1):
    print(f"{idx}. 红骰子：{roll[0]}，蓝骰子：{roll[1]}，绿骰子：{roll[2]}")

代码说明 #

• 使用itertools.product可以高效生成三个骰子的全部216种可能组合，避免手动构造列表的错误。
• 通过列表推导式精准筛选出和为5的组合，最终会得到6种符合条件的结果：(1,1,3)、(1,3,1)、(3,1,1)、(1,2,2)、(2,1,2)、(2,2,1)。
• 格式化后的输出清晰展示每一组符合条件的骰子点数组合，无多余异常输出。

内容的提问来源于stack exchange，提问作者Beatriz Mcknight