import numpy as np
from matplotlib.widgets import LassoSelector
import matplotlib.pyplot as plt

class SelectFromCollection:
    def __init__(self, ax):
        self.lasso = LassoSelector(ax, onselect=self.onselect, useblit=True)
    def onselect(self, verts):
        pass


if __name__ == '__main__':
    fig, ax = plt.subplots()

    np.random.seed(19680801)
    Z = np.random.rand(1095, 100)
    x = np.arange(-0.5, 100, 1)  # len = 11
    y = np.arange(4.5, 1100, 1)  # len = 7

    ax.pcolormesh(x, y, Z)
    # data2d = np.random.randint(0, 100,(1000,1000))
    # ax.imshow(data2d)
    selector = SelectFromCollection(ax)
    plt.show()