for i in range(2000):
    current = array[:, (i)]
if mask[current[0], current[1]] > 0:
    cols.append(i)
expected = array[:, (cols)]
assert np.allclose(result, expected)
removing entries from a numpy array