python - python 在 matplotlib.image.AxesImage 对象上绘制峰值

Question

我有一个AxesImage来自 pylab 的 Python 对象。如何在情节顶部绘制点？

例如，我imshow在我拥有的二维数组上做了，返回 AxesImage。然后我做了一些峰值发现并找到(i, j)了对应于峰值的对。现在我所要做的就是将它们覆盖在图像的顶部。

我认为这个scatter()函数通常是你如何绘制这样的东西（？）但我无法让它覆盖。

谢谢！

Answer 1

解决方案相当简单，但不知道您可以像这样使用 Axes 对象：

import matplotlib.pyplot as plt

# detect peaks somehow
i, j = detect_peaks(array2d)

# plot
fig, ax = plt.subplots()
ax.imshow(array2d)
ax.scatter(i, j)
plt.show()

对于大多数 matplotlib 专家来说可能非常简单，但我做了很多猜测。