if在函数定义中的语法错误

Question

我有这段代码：

def rho(switch,y,h):
    if switch=1:
        return 1.255
    if switch=2:
        return 1.255*np.exp(-y/h)

显然if语句有语法错误，我无法弄清楚它是什么。

Answer 1

def rho(switch,y,h):
    if switch == 1:
        return 1.255
    if switch == 2:
        return 1.255*np.exp(-y/h)

==代替=语句中的if