左侧二元运算符的 numpy 强制问题

Question

我正在实现一个类似数组的对象，它应该可以与标准 numpy 数组互操作。我刚刚遇到了一个恼人的问题，范围缩小到以下内容：

class MyArray( object ):
  def __rmul__( self, other ):
    return MyArray() # value not important for current purpose

from numpy import array
print array([1,2,3]) * MyArray()

这会产生以下输出：

[<__main__.MyArray instance at 0x91903ec>
 <__main__.MyArray instance at 0x919038c>
 <__main__.MyArray instance at 0x919042c>]

显然，而不是像我一样调用 MyArray().__rmul__( array([1,2,3]) )希望，为数组的每个单独元素调用 __rmul__ ，并将结果包装在对象数组中。在我看来，这不符合 python 的 强制规则。更重要的是，它使我的左乘法变得毫无用处。

有人知道解决这个问题的方法吗？

（我认为 a 可以使用 __coerce__ 来修复它，但链接的文档解释说，不再调用该 a 来响应二元运算符......）

Answer 1

事实证明 numpy 为这个问题提供了一个简单的解决方案。以下代码按预期工作。

class MyArray( object ):
  __array_priority__ = 1. # <- fixes the problem
  def __rmul__( self, other ):
    return MyArray()

更多信息可以在此处找到。