python - numpy python 类调用错误的 __setitem__

Question

我有以下类(class):

class autoArray2(numpy.ndarray):
    def __new__(self, *args, **kwargs):
        obj = numpy.array(*args, **kwargs)
        return(obj)

    def __setitem__(self, coords, value):
        print("HERE")

但是似乎调用的是 array.__setitem__ 而不是我指定的那个。

a = numpy.array([[1,2],[2,3]])
b = autoArray2(a)
a[0,0] = 1

没有打印“HERE”。

Answer 1

子类化 numpy 数组有点棘手。 Stefan van der Walt's slides和 numpy docs如果你想继承，是很好的起点。

import numpy as np

class AutoArray2(np.ndarray):
    def __new__(cls, input_array):
        # Input array is an already formed ndarray instance
        # We first cast to be our class type
        obj = np.asarray(input_array).view(cls)
        return obj
    def __array_finalize__(self, obj):
        if obj is None: return
    def __setitem__(self, coords, value):
        print("HERE")

a = np.array([[1,2],[2,3]])
b = AutoArray2(a)
b[0,0] = 1

产量

HERE

关键要素是调用view(cls)。没有它，您将返回一个普通的 ndarray，而不是 AutoArray2 实例。

此外，a[0,0] = 1 使用的是 a —— 普通的 ndarray。要使用 b 的 __setitem__，您需要 b[0,0] = 1。