from matplotlib import pyplot as plt


class Plot2D:

    def __init__(self, *args, **kwargs):
        self.plot = plt.imshow(*args, **kwargs)

    def set(self, value):
        self.plot.set_data(value)