Построение поля в полярных координатах

Question

Этот код может генерировать векторное поле в 2D. Я получил это здесь

import matplotlib.pyplot as plt
import numpy as np

X, Y = np.meshgrid(np.arange(0, 2 * np.pi, .2), np.arange(0, 2 * np.pi, .2))
U = np.cos(X)
V = np.sin(Y)
fig2, ax2 = plt.subplots()
ax2.set_title("pivot='mid'; every third arrow; units='inches'")
Q = ax2.quiver(X[::3, ::3], Y[::3, ::3], U[::3, ::3], V[::3, ::3],
               pivot='mid', units='inches')
qk = ax2.quiverkey(Q, 0.9, 0.9, 1, r'$1 \frac{m}{s}$', labelpos='E',
                   coordinates='figure')
ax2.scatter(X[::3, ::3], Y[::3, ::3], color='r', s=5)

Я хочу создать вектор электрического поля в полярных координатах (3D). Но вектор поля будет в декартовых координатах. У меня есть все данные (я имею в виду электрическое поле в x и y, и в направлении z, для разных радиусов, высот и углов). Есть ли способ преобразовать этот код в это?

Или, если у вас есть другие предложения, пожалуйста, дайте мне.