如何在Numpy中使用余切（cotangent）函数？

Numpy确实没有内置余切（cotangent）函数，但可以基于已有三角函数直接实现，核心依托余切的数学定义：cot(x) = cos(x) / sin(x)，也等价于1 / tan(x)。

基础实现方式 #

直接组合Numpy内置的三角函数即可：

import numpy as np

def cot(x):
    return np.cos(x) / np.sin(x)

或者通过正切函数实现：

import numpy as np

def cot(x):
    return 1 / np.tan(x)

处理特殊值 #

当x为π的整数倍时，sin(x)等于0，会触发除以0的警告或错误。可以针对这类特殊点做处理，返回数学上符合定义的无穷大：

import numpy as np

def cot(x):
    sin_x = np.sin(x)
    # 临时屏蔽除以0的警告
    with np.errstate(divide='ignore'):
        result = np.cos(x) / sin_x
    # 将sin(x)为0的位置设为inf（余切在这些点无定义）
    result[sin_x == 0] = np.inf
    return result

测试示例：

x = np.array([0, np.pi/4, np.pi/2, np.pi])
print(cot(x))
# 输出：[inf  1.  0. inf]

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