Python find local maxima 2d array