基于2.3节中随机游走的例子,使用ndarray和随机数生成函数模拟一个物体在三维空间随机游走的过程。
① 创建3×10的二维数组,记录物体每一步在三个轴向上的移动距离。在每个轴向的移动距离服从标准正态分布(期望均值为0,方差为1)。行序0、1、2分别对应x、y和z轴;
② 计算每一步走完后物体在三维空间的位置;
③ 计算每一步走完后物体距离原点的距离;
④ 统计物体在z轴上到达的最远距离;(提示:使用abs()绝对值函数对z轴每一步运动后的位置求绝对值,然后求最大距离)
⑤ 统计物体在三维空间距离原点的最近距离值
import numpy as np
rwlk=np.random.normal(0,1,(3,10))
rwlk=np.where(rwlk>0,1,-1)
#计算每一步走完后物体在三维空间的位置
post=rwlk.cumsum(axis=1)
# 计算每一步走完后物体距离原点的距离
dis0=np.sqrt(np.square(post[0])+np.square(post[1])+np.square(post[2]))
#统计物体在z轴上到达的最远距离
abs(post[2]).max()
#统计物体在三维空间距离原点的最近距离值
dis0.min()