import numpy as np from scipy import ndimage mask = np.array( [ [0, 1, 1, 0, 0, 0], [0, 0, 1, 0, 1, 0], [1, 1, 0, 0, 0, 1], [0, 0, 0, 1, 1, 1], ], dtype=bool, )