python - 使用 matplotlib 绘制一个简单的 3d numpy 数组

Question

我想绘制三维 ODE 系统的数值方法的结果。我的输出采用以下形式(假设我们已经计算了三个步骤):

import numpy as np
v= np.array([[1,2,3], [4,5,6], [7,8,9]])

每个三元组中的第一个值是 x 坐标，第二个是 y 坐标，第三个是 z 坐标。

我想要在 3D 网格上绘制这些点的最简单有效的方法。问题似乎是数据的格式应类似于 np.array([[1,4,7], [2,5,8], [3,6,9]])。

Answer 1

您可以像这样在 3D 中绘制结果:

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

v= np.array([[1,2,3], [4,5,6], [7,8,9]])
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.plot(v[:,0],v[:,1],v[:,2])
plt.show()