matplotlib（5）-- 等高线contourf、contour；imshow() ; colorbar(); 3D数据处理

 1 import matplotlib.pyplot as plt
 2 import numpy as np
 3 
 4 #定义计算高度的函数
 5 def f(x,y):
 6     return (1 - x/2 + x**5 +y**3) * np.exp(-x**2 - y**2)
 7 
 8 n = 256
 9 x = np.linspace(-3, 3, n)
10 y = np.linspace(-3, 3, n)
11 #核心函数是plt.contourf()，但在这个函数中输入的参数是x,y对应的网格数据以及此网格对应的高度值
12 #因此我们调用np.meshgrid(x,y)把x,y值转换成网格数据
13 X, Y = np.meshgrid(x, y)
14 
15 #plt.contourf 与 plt.contour 区别：
16 # f：filled，也即对等高线间的填充区域进行填充（使用不同的颜色）
17 # contourf：将不会再绘制等高线（显然不同的颜色分界就表示等高线本身）
18 
19