3

将 numpy 数组通过右手运算符传递后，如何访问它的属性__rsub__？

我在 python 中编写了一个非常简单的类，它定义了两个函数：

class test(object):
    def __sub__(self,  other):
        return other

    def __rsub__(self,  other):
        return other

基本上他们也应该这样做。左手运算符__sub__按预期工作，但似乎 numpy 数组在右手运算符上被剥离了它的属性

from skimage import data
from skimage.color import rgb2gray
lena = data.lena()

grayLena = rgb2gray(lena)
t = test()

## overloaded - operator
left_hand = t - grayLena
print left_hand
# Output: 
#array([[ 0.60802863,  0.60802863,  0.60779059, ...,  0.64137412,
#    0.57998235,  0.46985725],
#  [ 0.60802863,  0.60802863,  0.60779059, ...,  0.64137412,
#    0.57998235,  0.46985725],
#  [ 0.60802863,  0.60802863,  0.60779059, ...,  0.64137412,
#    0.57998235,  0.46985725],
#  ..., 
#  [ 0.13746353,  0.13746353,  0.16881412, ...,  0.37271804,
#    0.35559529,  0.34377725],
#  [ 0.14617059,  0.14617059,  0.18730588, ...,  0.36788784,
#    0.37292549,  0.38467529],
#  [ 0.14617059,  0.14617059,  0.18730588, ...,  0.36788784,
#    0.37292549,  0.38467529]])

right_hand = grayLena - t
print right_hand
# Output:
# array([[0.6080286274509803, 0.6080286274509803, 0.6077905882352941, ...,
#   0.6413741176470589, 0.5799823529411765, 0.4698572549019608],
#  [0.6080286274509803, 0.6080286274509803, 0.6077905882352941, ...,
#   0.6413741176470589, 0.5799823529411765, 0.4698572549019608],
#  [0.6080286274509803, 0.6080286274509803, 0.6077905882352941, ...,
#   0.6413741176470589, 0.5799823529411765, 0.4698572549019608],
#  ..., 
#  [0.1374635294117647, 0.1374635294117647, 0.1688141176470588, ...,
#   0.3727180392156863, 0.35559529411764706, 0.34377725490196076],
#  [0.1461705882352941, 0.1461705882352941, 0.18730588235294118, ...,
#   0.3678878431372549, 0.37292549019607846, 0.3846752941176471],
#  [0.1461705882352941, 0.1461705882352941, 0.18730588235294118, ...,
#   0.3678878431372549, 0.37292549019607846, 0.3846752941176471]], dtype=object)

所以这两个操作的区别在于__rsub__接收一个 dtype=object 的数组。如果我只是设置这个数组的 dtype，一切都会正常工作。

但是，它仅适用于 . 之外的返回值__rsub__。在我里面我__rsub__只有垃圾，我无法转换回来，即如果我这样做

npArray = np.array(other,  dtype=type(other))

我得到一个类型的一维数组（在我的例子中是浮点数）。但是由于某种原因，形状信息丢失了。有没有人这样做或知道如何访问数组的原始属性（形状和类型）？

4

1 回答 1

1

我不确定ndarray's 机器内部的确切控制流程是什么，但是在您的情况下发生的事情或多或少是清楚的：

ndarray委托给对象__rsub__方法的不是整体减法操作，而是从数组中的每个项目中减去对象。显然，当它必须将操作委托给对象的方法时，object无论返回什么，都将返回类型设置为。您可以通过对代码稍作修改来检查它：

class test(object):
    def __sub__(self,  other):
        return other

    def __rsub__(self,  other):
        return other if other != 1 else 666

In [11]: t = test()

In [12]: t - np.arange(4)
Out[12]: array([0, 1, 2, 3])

In [13]: np.arange(4) - t
Out[13]: array([0, 666, 2, 3], dtype=object)

我认为没有一种简单的方法可以覆盖这种行为。您可以尝试使用高和滥用方法创建test一个子类ndarray：__array_priority____array_wrap__

class test(np.ndarray):
    __array_priority__ = 100

    def __new__(cls):
        obj = np.int32([1]).view(cls)
        return obj

    def __array_wrap__(self, arr, context) :
        if context is not None :
            ufunc = context[0]
            args = context[1]
            if ufunc == np.subtract :
                if self is args[0] :
                    return args[1]
                elif self is args[1] :
                    return args[0]
        return arr

现在：

>>> t = test()
>>> np.arange(4) - t
array([0, 1, 2, 3])
>>> t - np.arange(4)
array([0, 1, 2, 3])

但：

>>> np.arange(4) + t
test([1, 2, 3, 4])
>>> t + np.arange(4)
test([1, 2, 3, 4])

这有点浪费，因为我们正在执行将1内部添加t到数组中的每个值的操作，然后默默地丢弃它，但我想不出任何方法来覆盖它。

于 2013-03-11T19:56:45.533 回答