如何在matplotlib imshow等高线图上叠加显示自定义数据点？

阿华AIGC实验室

2026-5-11

在imshow热图上添加实心圆标记新点的解决方案

没问题！要在你现有的imshow热图上标记x_new、y_new对应的z_new值，只需要用Matplotlib的scatter()函数就能轻松实现，还能让标记点的颜色和原图的色阶保持一致，直观体现新点的z值和周边区域的关系。

具体实现要点：

用plt.scatter()绘制实心圆：指定x_new和y_new作为坐标，marker='o'表示实心圆，s参数可以调整点的大小（比如s=100）。
让标记点颜色匹配原图：将c参数设为z_new，同时指定和imshow相同的cmap、vmin和vmax，这样标记点的颜色会和热图的色条对应。
可选优化：添加白色边缘色edgecolor='white'，让标记点在复杂的热图背景下更突出。

修改后的完整代码：

import numpy as np
import matplotlib.pyplot as plt
from matplotlib import colors

x = np.linspace(-0.5, 2.0, 101)
y = np.linspace(-0.5, 2.0, 101)
z = np.zeros((101, 101))
E = 0.0

for i in range(len(x)):
    for j in range(len(y)):
        z[i,j] = -max(y[j]+0.2+E, 0.5-x[i], 0)

x_new = np.array([1.1168189, 0.8381589, 1.3312789, -0.2149011])
y_new = np.array([1.7571379, 1.5555579, 1.9138179, 0.7912879])
z_new = []

for k, l in zip(x_new, y_new):
    z_new.append(-max(l+0.2+E, 0.5-k, 0))

fig, ax = plt.subplots()
ms = plt.imshow(z.T, cmap='plasma', vmin=-2.5, vmax=0, origin='lower', interpolation='none', extent=[-0.5,2.0,-0.5,2.0])
ax.set_xlabel('x', fontsize=16, fontname = "Helvetica")
ax.set_ylabel('y', fontsize=16, fontname = "Helvetica")
cbar = plt.colorbar(ms)
cbar.ax.tick_params(labelsize=10, direction='out')
cbar.set_label('z', fontsize=16, fontname = "Helvetica")
plt.xticks(fontname = "Helvetica", fontsize=12)
plt.yticks(fontname = "Helvetica", fontsize=12)

# 添加实心圆标记新点
plt.scatter(x_new, y_new, c=z_new, cmap='plasma', vmin=-2.5, vmax=0, marker='o', s=100, edgecolor='white')

plt.show()