i, j = np.mgrid[0 : pe.shape[0], 0 : pe.shape[1]]