问题描述
我正在从 numpy 扩展我自己的数据类型,几乎一切正常。我有 __array_ufunc__
和 __array_finalize__
可以将所有通用函数转换为我自己的类型。但是,对于 argmin 和 argmax,不会调用 __array_ufunc__
、 __array_finalize__
或 __array_wrap__
。应该使用什么机制来扩展这些功能,为什么没有发生这种情况?
解决方法
我刚刚意识到 argmin 和 argmax 使用 df1:
0 1
0 A A
1 A B
2 A D
df2:
0 1
0 A C
1 A A
2 A D
机制而不是通用函数一。