Python 随机漫步

Question

随机漫步：从（0，0）开始，走100步（x轴方向）。步进为1和－1（y轴方向），其两种步进发生的概率相等。（请使用random模块，for循环，if判断和列表来解决。）
1）生成类似下面的随机漫步图

2）求漫步中的最大值，最小值，和除去（0，0）点外，第一次y＝0的位置（如果没有这样的位置，则答案为None）。（仅可以使用randint、range、append、plot、min、max、index、print来解题。）

Answer 1

import matplotlib.pyplot as plt
import random

x_axis=range(101)
y = 0
y_axis=[y]
for i in range(100):
    if random.randint(0,1):
        y += 1
    else:
        y -= 1
    y_axis.append(y)

plt.plot(x_axis,y_axis)
plt.show()

print(max(y_axis))
print(min(y_axis))
i = y_axis[1:].index(0)
print((i+1,0)) if i else print('None')