如何在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()

输出