xdim, ydim = data.shape
k = 2
a, b = np.meshgrid(list(range(ydim)), list(range(xdim)))
mask = b - a - k < 0
new_data = data[mask]
new_data2 = np.array(data)
new_data2[~mask] = 0
вырезание массива 2D numpy по индексу