python绘图

绘制函数图像
绘制函数 \(y=sin(x)\) 的图像

import numpy as np
import matplotlib.pyplot as plt
x = np.arange(0, 2*np.pi, 0.01)
y = np.sin(x)
plt.figure(1)
plt.plot(x, y)
plt.show()

绘制方程图像
绘制方程 \(x^2+2xy^2+3y^2=1\) 的图像

import numpy as np
import matplotlib.pyplot as plt

def draw():
    x=np.arange(-2,2,0.01)
    y=np.arange(-2,2,0.01)
    # 转化为网格
    x,y=np.meshgrid(x,y)
    z=np.power(x,2)+2*x*y+3*np.power(y,2)-1
    # 绘制等高线
    plt.contour(x,y,z,0)
    plt.show()

draw()