Python数组中索引状态的变化

Question

我从步行者那里走了几步，每走一步都会增加压力。我想确定每一步的开始和停止以及它发生的时间。到目前为止，这个代码非常适用于确定压力何时上升或下降到阈值以下（脚落地和抬起）。现在我在上面的\u阈值数组中有真/假状态。你知道吗

data = np.loadtxt('junk.csv',delimiter=',',skiprows=1)
Fz = data

sigma = 5                       
threshold = 30                  
above_threshold = gaussian_filter(Fz, sigma=sigma) > threshold

我是Python新手，我想知道是否有一种有效的方法来构建一个新数组，它告诉我状态何时从false变为true（第一步开始），然后从true变为false（第一步结束）的索引。以及，开始和停止下一步。你知道吗

提前谢谢， 斯科特

Answer 1

以下将为您提供变化点的索引：

ind = np.where(np.diff(above_threshold))[0] + 1

如果要将这些索引拆分为成对的索引，可以执行以下操作

np.vstack((ind[::2], ind[1::2])).T