2020
10-06
10-06
python numpy实现rolling滚动案例
相比较pandas,numpy并没有很直接的rolling方法,但是numpy有一个技巧可以让NumPy在C代码内部执行这种循环。这是通过添加一个与窗口大小相同的额外尺寸和适当的步幅来实现的。importnumpyasnpdata=np.arange(20)defrolling_window(a,window):shape=a.shape[:-1]+(a.shape[-1]-window+1,window)strides=a.strides+(a.strides[-1],)returnnp.lib.stride_tricks.as_strided(a,shape=shape,strides=stri...
继续阅读 >