python 直方图匹配_python-查找二维直方图的峰值

我制作了一些(x,y)数据的2d直方图,并得到了这样的图像：

我想要一种方法来获取将最大值存储在H中的点的(x,y)坐标.例如,在上图的情况下,它将是带有aprox坐标的两个点：(1090, 1040)和(1110,1090).

这是我的代码：

import numpy as np

import matplotlib.pyplot as plt

import matplotlib.cm as cm

from os import getcwd

from os.path import join, realpath, dirname

# Path to dir where this code exists.

mypath = realpath(join(getcwd(), dirname(__file__)))

myfile = 'datafile.dat'

x, y = np.loadtxt(join(mypath,myfile), usecols=(1, 2), unpack=True)

fig = plt.figure()

ax = fig.add_subplot(111)

xmin, xmax = min(x), max(x)

ymin, ymax = min(y), max(y)

rang = [[xmin, xmax], [ymin, ymax]]

binsxy = [int((xmax - xmin) / 20), int((ymax - ymin) / 20)]

H, xedges, yedges = np.histogram2d(x, y, range=rang, bins=binsxy)

extent = [yedges[0], yedges[-1], xedges[0], xedges[-1]]

cp = ax.imshow(H.transpose()[::-1], interpolation='nearest', extent=extent, cmap=cm.jet)

fig.colorbar(cp)

plt.show()

编辑

我尝试了Marek和qarma发布的解决方案,试图获取垃圾箱的坐标而不是它们的索引,如下所示：

# Marek's answer

x_cent, y_cent = unravel_index(H.argmax(), H.shape)

print('Marek')

print(x_cent, y_cent)

print(xedges[x_cent], yedges[y_cent])

# qarma's answer

idx = list(H.flatten()).index(H.max())

x_cent2, y_cent2 = idx / H.shape[1], idx % H.shape[1]

local_maxs = np.argwhere(H == H.max())

print('

qarma')

print(x_cent2, y_cent2)

print(xedges[x_cent2], yedges[y_cent2])

print(xedges[local_maxs[0,0]], yedges[local_maxs[0,1]], xedges[local_maxs[1,0]], yedges[local_maxs[1,1]])

结果是：

Marek

(53, 50)

(1072.7838144329899, 1005.0837113402063)

qarma

(53, 50)

(1072.7838144329899, 1005.0837113402063)

(1072.7838144329899, 1005.0837113402063, 1092.8257731958763, 1065.3611340206187)

因此,最大坐标是相同的,这很好！现在我有一个小问题,因为如果放大2d图,我会发现全局最大值和局部最大值的坐标都稍微偏离中心：

为什么是这样？