[python] Peak detection in a 2D array