python绘制等高线和等值面初步

import matplotlib.pyplot as plt
import numpy as np
 
x = np.linspace(-5, 6, 210)
y = np.linspace(-5, 6, 210)
x, y = np.meshgrid(x, y)
z = (1 - x/2 + x**5 + y**4) * np.exp(-x**2 - y**2)
 
plt.contour(x, y, z, levels=9, colors='black')
 
plt.show()

 

import matplotlib.pyplot as plt
import numpy as np
 
x, y = np.meshgrid(np.linspace(-4, 4, 500), np.linspace(-4, 4, 500))
z = (x ** 2 + y ** 2) * np.exp(-(x ** 2 + y ** 2) / 3)

plt.contourf(x, y, z, 7, cmap='jet')
 
C = plt.contour(x, y, z, 7, colors='black', linewidths=0.5)
plt.clabel(C, inline=True, fontsize=10)
 
plt.show()

 

Matplotlib 是一个 Python 的 2D绘图库；
NumPy 是一个 Python 包， 它代表 “Numeric Python”，它是一个由多维数组对象和用于处理数组的例程集合组成的库；
plt.contour是 Matplotlib 库中的一个函数,用于绘制等高线图；

先初步了解一下；