增加数组维度
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()