等高线图绘制
import matplotlib.pyplot as plt
import numpy as np
def f(x, y):
#The height function
return (1 - x/2 + x**5 + y**3) * np.exp(-x**2-y**2)
n = 256
x = np.linspace(-3, 3, n)
y = np.linspace(-3, 3, n)
X, Y = np.meshgrid(x, y)
# Use plt.contourf to filling contours 等高线
# X, Y and value for (X,Y) point
plt.contourf(X, Y, f(X, Y), 8, alpha=0.75, cmap= plt.cm.hot)
# Use plt.contourf to add contour lines 黑线
C = plt.contour(X, Y, f(X, Y), 8, colors='black')
# adding label 标签
plt.clabel(C, inline=True, fontsize= 10)
plt.xticks(())
plt.yticks(())
plt.show()
ZhengJohn 发布了25 篇原创文章 · 获赞 0 · 访问量 766 私信 关注