python - 为什么 numpy.angle() 不是 ufunc？

Question

为什么numpy.angle()不是 numpy 通用函数 (ufunc)？

它似乎符合 numpy documentation 中 ufunc 的标准但它没有被列为其中之一。

我认为它可能不符合定义，因为它转换数字的类型(从复数到实数)，但已经有其他 ufunc 可以执行此操作(例如 np.absolute)。

我遇到这个问题是因为我试图将np.angle直接应用于包含复数的xarray DataArray，并且它返回一个numpy数组而不是xarray DataArray。我认为它这样做是因为它不是 numpy ufunc，并且 xarray 会检查这一点。

Answer 1

np.angle 的完整代码(来自 numpy GitHub repo)

def angle(z, deg=0):
    if deg:
        fact = 180/pi
    else:
        fact = 1.0
    z = asarray(z)
    if (issubclass(z.dtype.type, _nx.complexfloating)):
        zimag = z.imag
        zreal = z.real
    else:
        zimag = 0
        zreal = z
    return arctan2(zimag, zreal) * fact

所以它是 np.arctan2 的一个瘦 Python 包装器，它是一个 ufunc。

一个猜测是，没有人认为它需要成为一个 ufunc，或者有兴趣将其转换为这样的工作。

https://docs.scipy.org/doc/numpy-1.15.1/reference/c-api.ufunc.html

https://docs.scipy.org/doc/numpy-1.15.1/reference/c-api.generalized-ufuncs.html

https://docs.scipy.org/doc/numpy-1.15.0/user/c-info.ufunc-tutorial.html

在我看来，直接使用 np.arctan2 以及您自己的复杂 dtype 数组的瘦包装器会简单得多。