python增加数组维度与用matplotlib画三维轨迹图

增加数组维度

x=position[0]
x=np.expand_dims(x,axis=0)
y=position[1]
y=np.expand_dims(y,axis=0)
z=position[2]
z=np.expand_dims(z,axis=0)

画三维轨迹图

#2.基于2.3节中随机游走的例子，使用ndarray和随机数生成函数模拟一个物体在三维空间
#随机游走的过程。
from mpl_toolkits.mplot3d import axes3d
import matplotlib.pyplot as plt
import numpy as np
np.set_printoptions( precision = 4)   #只显示4位小数
#创建3×10的二维数组，记录物体每一步在三个轴向上的移动距离。在每个轴向的移动距
#离服从标准正态分布（期望为0，方差为1）。行序0、1、2分别对应x、y和z轴；
rndwlk = np.random.normal(0, 1, size = (3,10))
print(rndwlk)

#计算每一步走完后物体在三维空间的位置；
position = rndwlk.cumsum(axis = 1)
print(position)

#画出三维轨迹
x=position[0]
x=np.expand_dims(x,axis=0)
y=position[1]
y=np.expand_dims(y,axis=0)
z=position[2]
z=np.expand_dims(z,axis=0)

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

# Plot a basic wireframe.
ax.plot_wireframe(x, y, z, rstride=10, cstride=10)
plt.show()