Python实现Σ求和时如何将循环条件作为可执行参数而非字符串传入？

现存问题点 #

• 原代码存在语法错误：else分支下的condition for i in range(i,target+1):是无效语法，无法正常执行
• 逻辑设计误区：传入的condition参数是字符串类型，Python不会自动将字符串解析为运算逻辑；举例的condition='i'*2实际是字符串拼接操作，生成普通字符串'ii'，和预期的i*2运算逻辑完全无关
• 变量作用域混淆：函数入参i和循环遍历用的i重名，会导致逻辑出错

推荐修复方案（无安全风险） #

放弃传递字符串条件的设计，直接把运算逻辑封装为函数（比如lambda匿名函数）作为参数传入，从根源上避免字符串解析问题，符合Python设计规范：

def sigma(start, end, calc_func=lambda x: x):
    total = 0
    for num in range(start, end + 1):
        total += calc_func(num)
    print(total)

调用示例 #

• 计算1到100的整数和（默认逻辑，和原代码condition='i'的效果一致）：sigma(1, 100)
• 计算1到100每个数乘2后的累加和：sigma(1, 100, calc_func=lambda x: x * 2)
• 计算1到100每个数平方的累加和：sigma(1, 100, calc_func=lambda x: x ** 2)

字符串表达式兼容方案（仅可信输入场景使用） #

如果一定要保留传入字符串条件的用法，需要用eval手动指定变量作用域解析表达式，注意：eval会执行任意传入的代码，仅可用于完全可信的输入，禁止用于接收外部不可信用户的输入
代码实现如下：

def sigma(start, end, condition='i'):
    total = 0
    for i in range(start, end + 1):
        # 绑定当前循环变量i的作用域，解析condition字符串
        total += eval(condition, {'i': i})
    print(total)

调用注意事项 #

• 计算1到100每个数乘2的和时，需要传入字符串形式的表达式'i*2'，而非'i'*2（后者会生成字符串'ii'）：
  sigma(1, 100, 'i*2')

内容的提问来源于stack exchange，提问作者Japjot Singh Chauhan