Python多if分支计算函数仅首个分支生效，返回错误结果求修正

• 变量赋值逻辑错误：你给出的参数示例中使用了比较运算符==而非赋值运算符=，没有实际给direction和number设置你期望的'B'、2值，导致分支判断时匹配失败。同时字符串值B需要包裹引号，否则会被识别为未定义的变量报错。
• 函数设计不合理：当前函数直接依赖全局变量做判断，很容易因为变量作用域问题、值被意外篡改导致逻辑异常，建议将判断变量作为参数传入函数。

# 正确为变量赋值
direction = 'B'
number = 2 

# 改造函数接收参数，消除全局变量依赖
def price_cal(direction, number):
    if direction == 'A' and number == 1:
        return 1
    elif direction == 'A' and 1 < number <= 9:
        return 2
    elif direction == 'A' and 9 < number <= 13:
        return 3
    elif direction == 'B' and number == 1:
        return 4 
    elif direction == 'B' and 1 < number <= 9:
        return 5
    else:
        return 6

# 调用函数时传入对应参数
print(price_cal(direction, number))

上述代码运行后输出5，和你预期的返回值一致。

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