matplotlib绘制极坐标图（双纽线） 和3d图

废话不多说直接上代码

from matplotlib import pyplot as plt
import numpy as np

fig,axe = plt.subplots(figsize=(10,10),dpi=80)

# 双纽线 极坐标方程：r=a**2*cos(2*X)
def polar():
    x = np.arange(0,2*np.pi,0.01)
    y = [2*np.cos(2*i) for i in x]
    # 设置projection参数为极坐标polar
    # 也可以用  plt.polar(x,y)来绘制极坐标中的图像
    axe = plt.subplot(projection='polar')
    axe.plot(x, y)
    
    plt.show()

# 绘制三维图形
def matplot3D():
    # 引入相关的包
    from mpl_toolkits.mplot3d import Axes3D
	# 这里需要传入fig，否则会报错
    axe = Axes3D(fig)
    x = np.random.rand(60)
    y = np.random.rand(60)
    z = np.random.rand(60)
    a = np.random.rand(60)
    # 绘制3d折线图和3d散点图
    axe.plot(x,y,a,color='r')
    axe.scatter(x,y,z)
    plt.show()

matplot3D()