Python数学游戏代码精简优化求助（新手入门）

一、替换重复的if-elif分支 #

• 用字典映射实现分支逻辑：把运算类型与对应的计算逻辑、数字生成逻辑绑定，避免冗长的条件判断。这种方式更易扩展，新增运算只需在字典中添加条目即可。

二、复用数字生成逻辑 #

• 封装通用生成函数：将不同运算的数字生成逻辑拆分为独立函数，或编写带参数的通用函数，通过传入参数适配不同运算的数字范围规则。

三、从零优化的学习方向 #

• 掌握函数式编程基础：了解lambda表达式、字典映射的用法，用函数作为值来简化逻辑。
• 学习代码重构技巧：重点练习「提取重复代码为函数」「用多态/映射替换条件分支」这两类核心重构手法。
• 入门面向对象思想（可选）：如果后续要扩展游戏功能（比如添加除法、自定义难度），可以将每种运算封装为类，通过多态统一调用接口，彻底消除条件分支。

重构示例 #

原冗余代码片段 #

import random

op = random.choice(['+', '-', '*'])
if op == '+':
    a = random.randint(1, 100)
    b = random.randint(1, 100)
    result = a + b
elif op == '-':
    a = random.randint(10, 200)
    b = random.randint(1, a)
    result = a - b
elif op == '*':
    a = random.randint(1, 20)
    b = random.randint(1, 20)
    result = a * b

优化后代码 #

import random

# 拆分数字生成逻辑为独立函数
def generate_add_nums():
    return random.randint(1, 100), random.randint(1, 100)

def generate_sub_nums():
    a = random.randint(10, 200)
    b = random.randint(1, a)
    return a, b

def generate_mul_nums():
    return random.randint(1, 20), random.randint(1, 20)

# 用字典映射绑定运算与对应的逻辑
op_handlers = {
    '+': (generate_add_nums, lambda a, b: a + b),
    '-': (generate_sub_nums, lambda a, b: a - b),
    '*': (generate_mul_nums, lambda a, b: a * b)
}

# 统一调用逻辑
selected_op = random.choice(list(op_handlers.keys()))
num_generator, calculator = op_handlers[selected_op]
num1, num2 = num_generator()
result = calculator(num1, num2)

内容的提问来源于stack exchange，提问作者Jacob Graves