首页 > 解决方案 > 为字符串 dtypes 写一个 `__array_ufunc__`

问题描述

我正在实现一个混合NDArrayOperatorsMixin使用此处描述的方法的类。

这适用于数字,但不适用于字符串 dtype。例如,

x = MyNewArrayClass(np.array(["a", "b", "c"]))  
x == "a"

提出以下问题UFuncTypeError

numpy.core._exceptions.UFuncTypeError: ufunc 'equal' did not contain a loop with signature matching types (dtype('<U1'), dtype('<U1')) -> dtype('bool')

如何修改文档中建议的实现以支持 str dtypes?

标签: numpynumpy-ndarray

解决方案


推荐阅读