python - matplotlib 轮廓 : get Z value under cursor

Question

当我使用 contourf 绘制某些东西时，我会在绘图窗口的底部看到鼠标光标下的当前 x 和 y 值。有没有办法同时查看 z 值？

这里有一个例子contourf:

import matplotlib.pyplot as plt
import numpy as hp
plt.contourf(np.arange(16).reshape(-1,4))

Answer 1

显示光标位置的文本由 ax.format_coord 生成。您可以覆盖该方法以同时显示 z 值。例如，

import matplotlib.pyplot as plt
import numpy as np
import scipy.interpolate as si
data = np.arange(16).reshape(-1, 4)
X, Y = np.mgrid[:data.shape[0], :data.shape[1]]
cs = plt.contourf(X, Y, data)

func = si.interp2d(X, Y, data)
def fmt(x, y):
    z = np.take(func(x, y), 0)
    return 'x={x:.5f}  y={y:.5f}  z={z:.5f}'.format(x=x, y=y, z=z)


plt.gca().format_coord = fmt
plt.show()