如何在 Python Matplotlib 中绘制密度图？

要在 Python 中绘制密度图，我们可以按照以下步骤操作——

  • 使用 numpy 创建 side、x、y 和 z。Numpy linspace 有助于根据第三个数在两个点之间创建数据。

  • 使用 side 数据从坐标向量返回坐标矩阵。

  • 使用 x 和 y（步骤 2）创建指数数据。

  • 使用 pcolormesh() 方法创建具有非规则矩形网格的伪彩色图。

  • 要显示图像，请使用 show() 方法。

实例

from matplotlib import pyplot as plt, cm, colors
import numpy as np

plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True

side = np.linspace(-2, 2, 15)
X, Y = np.meshgrid(side, side)
Z = np.exp(-((X - 1) ** 2 + Y ** 2))

plt.pcolormesh(X, Y, Z, shading='auto')

plt.show()

输出

更新于： 2021-9-29

3 千+ 浏览

开启你的 职业生涯

完成课程获得认证

开始学习
广告