Sympi中计算复数辐角arg(0)返回NaN的原因及设置arg(0)=0的方法

问题原因 #

复数0在复平面上没有确定的方向，其辐角属于数学上的无定义值，SymPy原生arg()函数严格遵循该数学规范，因此输入0时会返回NaN（非数值）表示结果无意义。

解决方案 #

1. 自定义封装函数（最常用） #

对原生arg()做一层分支判断，单独处理0输入的场景：

import sympy as sp

def my_arg(z):
    # 兼容符号场景下的0判断
    if sp.simplify(z == 0):
        return sp.Integer(0)
    return sp.arg(z)

调用my_arg(0)即可返回0，其余非0输入的计算逻辑和原生arg()完全一致。

2. 符号表达式分支定义 #

如果需要在符号运算链中嵌入该逻辑，可以用Piecewise定义带分支的辐角计算规则：

z = sp.symbols('z')
arg_rule = sp.Piecewise(
    (0, z == 0),
    (sp.arg(z), True)
)

后续代入数值计算时，arg_rule.subs(z, 0)会返回0，其他输入正常计算辐角。

3. 结果后处理 #

如果已经完成批量辐角计算，可以直接替换结果中的NaN为0：

result = sp.arg(0)
result = 0 if sp.isnan(result) else result

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