如何让Python内置complex()函数兼容math模块函数？无需numpy实现复数运算

• 完全不需要依赖numpy，Python标准库自带的cmath模块就是专门用来处理复数的数学运算的，涵盖三角函数、指数、对数等常用操作。
• 你写的代码报错是因为math模块下的函数仅支持实数输入，传入复数类型会直接抛出TypeError。

把你的代码改成用cmath就能正常运行：

import cmath

def f(a, b):
    return cmath.cos(complex(a, b))

print(f(1, 1))

运行后会输出复数形式的余弦结果，比如(0.8337300251311491-0.9888977057628651j)。

另外，Python内置的复数类型本身也支持乘、除、幂运算，比如：

z1 = complex(1, 1)
z2 = complex(2, 3)
print(z1 * z2)  # 输出 (-1+5j)
print(z1 ** 2)  # 输出 (0+2j)

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