在matplotlib中绘制一个基于截锥体的涡旋

Question

我试图在我的数学论文中为一个冠状孔做一个基本模型。我想创建一个图，其中圆柱体的径向方向与高度成比例地增加，然后我想在$（\ theta，z）$方向上明确地添加扭曲或者沿着曲面添加线和箭头。

目前我只能生产正常的气瓶。

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

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# Cylinder
x=np.linspace(-1, 1, 100)
z=np.linspace(-2, 2, 100)
Xc, Zc=np.meshgrid(x, z)
Yc = np.sqrt(1-Xc**2)

# Draw parameters
rstride = 20
cstride = 10
ax.plot_surface(Xc, Yc, Zc, alpha=0.2, rstride=rstride, cstride=cstride)
ax.plot_surface(Xc, -Yc, Zc, alpha=0.2, rstride=rstride, cstride=cstride)

ax.set_xlabel("X")
#ax.set_ylabel("Y")
fig.gca().set_ylabel(r'$\theta$')
ax.set_zlabel("Z")
plt.figure(figsize=(200,70)) #sets the size of the plot
#plt.show()