Numpy Inheritance; add a method to Numpy Array

Question

Let's say we have a 2D array, image, eg. 20x20. I would like add a method, called 'imshow' to this object such that whenever I do image.imshow(**kwargs)), the method imshow will make a call of Matplotlib.pyplot.imshow

What is the best way to do this? I was thinking of writing a class with an inheritance from numpy.ndarray, and adding a method 'imshow'.

Mehdi Rezaie · Accepted Answer · 2020-05-13 03:02:57Z

1

Just found the answer (thanks to 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()

answered May 13, 2020 at 3:02

Mehdi Rezaie

3261 gold badge3 silver badges8 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Numpy Inheritance; add a method to Numpy Array

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related