通过前面对NumPy的学习,相信大家一定对NumPy这个科学计算包有了一定的了解,接下来,本节将通过酒鬼漫步的案例来介绍如何运用NumPy随机数模块与数据处理。
下面先为大家描述一下场景,在一片空旷的平地上(一个二维地面上)有一个酒鬼,他最初停留在原点的位置,这个酒鬼每走一步时,方向是不确定的,在经过时间t之后,我们希望计算出这个酒鬼与原点的距离。
例如,这个酒鬼走了2000步(每步为0.5米),向前走一步记为1,向后走一步记为-1,当计算距原点的距离时,就是将所有的步数进行累计求和。因此,使用random模块来随机生成2000个“掷硬币值”(两个结果任选一个),具体代码如下:
导入numpy包
import numpy as np
steps=2000
draws=np.random.randint(0, 2, size=steps