Numpy 继承；向 Numpy 数组添加方法

Question

假设我们有一个二维数组image，例如。 20x20。我想为这个对象添加一个名为“imshow”的方法，这样每当我做image.imshow(**kwargs))时，方法imshow都会调用Matplotlib.pyplot.imshow

最好的方法是什么？我正在考虑编写一个继承自 numpy.ndarray 的类，并添加一个方法“imshow”。

Answer 1

刚刚找到答案（感谢How can a class that inherits from a NumPy array change its own values?）！

class array(np.ndarray):

    def __new__(cls, a):
        obj = np.asarray(a).view(cls)
        return obj

    def imshow(self, **kwargs):
        plt.figure()
        plt.imshow(self, **kwargs)
        plt.show()