значения сетки сетки питона из кадра данных панд

Question

Решено

У меня есть пандас DataFrame, подобный этому

       X    Y       Z      Vx           Vy              Vz
0   -17.5   10.0   2.5     0.037132     -4.021866e-02   -0.002117
1   -17.5   12.5   0.0     0.021691     -3.838874e-02   -0.000651
3   -17.5   0.0    12.5    0.063864     7.724853e-07    -0.023265
7   -17.5   2.5    2.5     0.072648     -1.244435e-02   -0.002749
16  -17.5   5.0    0.0     0.064299     -2.709658e-02   -0.000972

, и я хотел бы построить трехмерный колчан, где X, Y, Zявляются декартовыми координатами, а Vx, Vy, Vz являются компонентами моего векторного поля.Я сделал сетку для x, y, z таким образом

x,y,z = np.meshgrid(dataPM[dataPM.columns[0]],dataPM[dataPM.columns[1]],dataPM[dataPM.columns[2]])

, но я изо всех сил пытаюсь создать функцию для создания соответствующей сетки для Vx, Vy, Vz.

Есть идеи?

Большое спасибо за помощь.

Answer 1

Решение

Оказывается, что если у вас уже есть DataFrame pandas со всеми значениями x, y, z, Vx, Vy, Vz, вы можете просто передать их вдрожь, я виноват, но, как обычно, чем больше ты пытаешься, тем больше ты учишься.

Таким образом, окончательный код был:

from mpl_toolkits.mplot3d import axes3d
fig = plt.figure(figsize=(16,9))
ax = fig.gca(projection='3d')
ax.quiver(dataPM[dataPM.columns[0]], dataPM[dataPM.columns[1]], dataPM[dataPM.columns[2]],dataPM[dataPM.columns[3]],dataPM[dataPM.columns[4]],dataPM[dataPM.columns[5]])
plt.show()

И затем, если ты хочешь сделать это лучше, ты можешь играть сцвета и т. д.

Спасибо всем