要画等高线,需要用到 plt.contourf()
,可以根据提供的数据生成等高线,并根据等高线下的区域填充不同的颜色。该函数通常用于可视化二维数据集的密度分布或二维函数的等高线图。它接受三个参数:X, Y 和 Z,其中X和Y分布是二维数据集中的x坐标和y坐标,Z是对应的数据值。
代码如下:
import matplotlib.pyplot as plt
import numpy as np
x = np.linspace(-2, 2, 100)
y = np.linspace(-2, 2, 100)
X, Y = np.meshgrid(x, y)
Z = np.sin(X) * np.cos(Y)
plt.contourf(X, Y, Z, cmap='viridis')
plt.colorbar()
plt.xlabel('X')
plt.ylabel('Y')
plt.title('Filled Contour Plot of sin(x) * cos(y)')
plt.show()
结果如下:
给个赞呗~