python 画隐函数图像 画三维显函数图像

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import numpy as np

#隐函数画法1
xrange = np.linspace(-2 ,2, 400)
yrange = np.linspace(-2, 2, 400)

X, Y = np.meshgrid(xrange,yrange)
R = Y**2 + X**2
plt.contour(X,Y,R,[1])
plt.show()

########### 隐函数画法2
from sympy.parsing.sympy_parser import parse_expr
from sympy import plot_implicit
ezplot = lambda exper: plot_implicit(parse_expr(exper))#用了匿名函数
expression='x**2+y**2-1'#隐函数是x**2+y**2-1=0，其实就是圆的方程
ezplot(expression)#能描绘大致的图像

########### 三维
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
fig = plt.figure()
ax = Axes3D(fig)
X = np.arange(-4, 4, 0.25)
Y = np.arange(-4, 4, 0.25)
X, Y = np.meshgrid(X, Y)
R = np.sqrt(X ** 2 + Y ** 2)
Z = np.sin(R)
ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap='hot')
plt.show()