以下是一个使用径向颜色映射在网格图上的示例代码:
import numpy as np
import matplotlib.pyplot as plt
# 生成一些示例数据
x = np.linspace(-10, 10, 100)
y = np.linspace(-10, 10, 100)
X, Y = np.meshgrid(x, y)
Z = np.sqrt(X**2 + Y**2) # 这里使用径向距离作为示例数据
# 创建一个图形窗口和子图
fig, ax = plt.subplots()
# 在子图上使用pcolormesh函数绘制网格图
cax = ax.pcolormesh(X, Y, Z, cmap='jet')
# 添加颜色条
cbar = fig.colorbar(cax)
# 显示图形
plt.show()
这个示例代码中,首先使用numpy生成了一些示例数据,然后使用matplotlib的pcolormesh函数在子图上绘制了一个网格图。在这里,我们将生成的数据作为Z轴的值,将x和y作为网格的坐标。通过指定cmap参数为'jet',我们使用了一个径向颜色映射。最后,通过调用fig.colorbar函数添加了一个颜色条来表示不同值对应的颜色。
你可以根据自己的具体需求,调整生成数据的方式和使用的颜色映射,来实现你想要的效果。