Python TypeError及带括号表达式解析异常问题排查

报错信息 #

TypeError: unsupported operand type(s) for +: 'float' and 'NoneType'

问题现象 #

• 输入带[]的表达式时，仅解析括号内内容，前后的运算符和数值完全不处理
• 输入包含两个[]的表达式（比如[1+2]+[3*4]）时，函数直接返回None

代码问题分析 #

1. extract_expression函数逻辑完全错误
   函数里的while ']' in s:循环根本没起作用，一进入就直接return；而且用s.index('[')找第一个左括号，s.rindex(']')找最后一个右括号，导致提取的子串是从第一个[到最后一个]之间的所有内容——比如输入[1+2]+[3*4]，会提取1+2]+[3*4，后续替换时根本找不到匹配的括号串，导致括号无法被正确替换，最终calculate函数处理到无效字符串时返回None。

2. calculate函数缺少默认返回值
   当遍历完所有运算符都没找到匹配（比如字符串是空，或者格式错误），函数没有明确返回值，Python会默认返回None。当这个None和其他float类型的计算结果一起参与运算时，就会触发TypeError。

3. 括号替换逻辑有漏洞
   原代码每次替换都会把所有匹配的[sub_expression]替换掉，但如果有多个相同的子表达式，会导致错误替换；更关键的是extract_expression提取错误的子串，导致替换操作完全失效。

修复后的代码 #

def divide(a, b):
    return a / b

def pow(a, b):
    return a ** b

def addA(a, b):
    return a + b

def subA(a, b):
    return a - b

def mul(a, b):
    return a * b

operators = {
    '+': addA,
    '-': subA,
    '*': mul,
    '/': divide,
    '^': pow,
}

def extract_expression(s):
    # 找到第一个左括号的位置
    start = s.index('[')
    # 从左括号位置开始，找对应的右括号（支持嵌套括号）
    bracket_count = 1
    end = start + 1
    while end < len(s) and bracket_count > 0:
        if s[end] == '[':
            bracket_count += 1
        elif s[end] == ']':
            bracket_count -= 1
        end += 1
    # 返回括号内的内容（不包含括号）
    return s[start + 1:end - 1]

def calculate(s):
    # 先处理所有括号，从最内层开始
    while '[' in s:
        sub_expr = extract_expression(s)
        result = calculate(sub_expr)
        # 只替换第一个匹配的括号对，避免错误替换多个相同子串
        s = s.replace('[' + sub_expr + ']', str(result), 1)
    
    # 处理字符串前后空格，避免空字符串干扰
    s = s.strip()
    # 处理纯数字的情况
    if s.isdigit():
        return float(s)
    
    # 按运算符优先级遍历（先加减后乘除，幂运算最后）
    for op in ['+', '-', '*', '/', '^']:
        left, operator, right = s.partition(op)
        if operator:
            # 递归计算左右两边的值
            left_val = calculate(left)
            right_val = calculate(right)
            return operators[operator](left_val, right_val)
    
    # 兜底处理：如果是单个浮点数（比如"3.14"）直接转换，避免返回None
    try:
        return float(s)
    except ValueError:
        # 无法解析时返回0.0，避免None引发类型错误
        return 0.0

calc = input("Type calculation:\n")
print("Answer: " + str(calculate(calc)))

关键修改说明 #

• extract_expression函数：现在能正确匹配与第一个左括号对应的右括号，支持嵌套括号（比如[[1+2]*3]+4），提取的是最内层的有效括号内容。
• calculate函数：
  • 添加了s.strip()处理字符串前后空格，避免空字符串导致的解析异常
  • 最后增加了兜底的浮点数转换和异常处理，确保函数永远不会返回None
  • 修改了运算符遍历顺序，尽量符合常规运算优先级
  • 替换括号时使用replace(..., 1)只替换第一个匹配项，避免错误替换多个相同子串

测试示例：

• 输入[1+2]+[3*4]，返回15.0
• 输入10/[2+3]，返回2.0
• 输入[[2^3]+4]*2，返回24.0

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