1. 绘制一个二维随机漫步的图形
import matplotlib.pyplot as plt
import numpy as np
nsteps = 1000
draws = np.random.randint(-1,2,size=(2,nsteps))
walks = draws.cumsum(1)
plt.plot(walks[0,:],walks[1,:]);
plt.show()
首先设定步数为1000,在坐标轴上,x方向可以前进、后退、或者不动,y方向也是这样,即x(-1,0,1),y(-1,0,1),每一步都有9种选择,使用cumsum函数求得每一步的结果,然后Plot出来。
具体函数的使用 |
random.randint:
low、high、size三个参数。默认high是None,如果只有low,那范围就是[0,low)。如果有high,范围就是[low,high)。如果没有给定size,则返回单一的一个整数;否则返回一个数组。
cumsum:
cumsum(1)是水平求和,cumsum(0)是垂直求和。