如何解决numpy从一维数组中选择多个范围
假设我有一个一维值数组:
T = np.array([1.3,8.9,1.4,3.2,4.4,7.0,2.0,6.9]
,我有一个起始索引列表:
I = np.array([5,2,4,1])
对于每个起始索引,我想从起始索引开始的m
中获取T
个连续值。因此,m=3
的最终输出应该是带有以下内容的2D数组:
T = np.array([[7.0,6.9],[1.4,4.4],[4.4,2.0],[8.9,3.2]])
我可以简单地遍历索引,但是我希望有一种更有效的方法:
m = 3
out = np.empty((len(I),m))
for j,i in enumerate(I):
out[j] = T[j : j+m]
解决方法
这是不需要安装其他依赖项的答案:
def rolling_window(a,window):
a = np.asarray(a)
shape = a.shape[:-1] + (a.shape[-1] - window + 1,window)
strides = a.strides + (a.strides[-1],)
return np.lib.stride_tricks.as_strided(a,shape=shape,strides=strides)
m = 3
rolling_T = rolling_window(T,m) # Note that this is a very efficient view
out = rolling_T[I]
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。