问题描述

酒鬼漫步

在一条东西走向的大马路上,酒鬼所处的初始位置假定为原点,酒鬼随机向东(正向)或向西(反向)走一步。请使用所学的知识解决以下问题:

  • 1)酒鬼随机走了1000步,打印输出酒鬼的走步。(可假定-1为向西走,+1为向东走,如:-1,-1,1,-1,1,…-1)
  • 2)打印输出,酒鬼任意一次走完后距离原点的步数。(如:-1,-2,-1,-2,-1,0,1,2,1…46)
  • 3)打印输出,酒鬼最后距离原点的步数。
  • 4)打印输出,酒鬼正向走的最远步数。
  • 5)打印输出,酒鬼反向走的最远步数。
  • 6)计算酒鬼第一次距离原点5步的时候(可以是正向5步,也可以是反向5步)。此时酒鬼共走了多少步?

分析

题目非常的简单,看到题目中有一个要求是计算任意一次走完后的距离原点的步数,就是累积求和函数,可以使用numpy库的相关方法实现,所以我们使用numpy库函数。

代码实现

考虑到数据量的问题,这里以 100 步进行演示。

import numpy as np
# choice函数接收一个迭代对象和大小,这里选择100次
lt = np.random.choice([1, -1], 100)

print('-----------------1.酒鬼的走步-------------------')
print(lt)
cum_sum = lt.cumsum()
print('--------2.酒鬼任意一次走完后距离原点的步数----------')
print(cum_sum)
su = lt.sum()
print('------------3.酒鬼最后距离原点的步数--------------')
print(su)
print('-------------4.酒鬼正向走的最远步数---------------')
# 返回的是下标,这里考虑到下标是从零开始的,所以加一
print(cum_sum.argmax() + 1)
print('-------------5.酒鬼反向走的最远步数---------------')
print(cum_sum.argmin() + 1)
count = 0
for i in cum_sum:
    count += 1
    if i == 5 or i == -5:
        break
print('--------6.第一次距离原点5步的时候走的步数-----------')
print(count)

执行结果: