Python:如何找到不同符号的值之间的时间间隔？

Question

你好，我想计算一个信号的一阶导数的两个零点的时间差。例如，假设一阶导数为：

d =[-0.2, 0.3, 0.2, 0.2, -0.1, 0.5]
t =[   0,  13,  22,  23,   34,  50]

我想要一个数组c

c = [(13-0), (34-13), (50-34)], i.e. c = [12, 21, 16]

假设我在pUser中为不同的user记录了原始数据。

z = list()
for i in user:
    sort_ind = np.argsort(pUser.time[i])
    time_vec = pUser.time[i][sort_ind]
    val_vec = pUser.val[i][sort_ind]
    tmp0 = np.diff(val_vec)   
    s = np.sign(tmp0[0])  ## Sign of the first value (+1 or -1)
    index = 0
    zz = list()
    for j in range(1,len(tmp0)):
        if (np.sign(tmp0[j]) + s)==0:
            s = np.sign(tmp0[j])
            dt = (time_vec[j]-time_vec[index])/(2*np.timedelta64(1, 's'))
            zz.append(dt)
            index = j
    z.append(zz)

但是，由于len(tmp0) ~ 10^4-10^5的原因，该循环需要一段时间。我想知道是否有更快的方法来避免这种循环。

Answer 1

一个纯粹的麻木的答案如何：

import numpy as np

a = np.array([[-0.2, 0.3, 0.2, 0.2, -0.1, 0.5],
              [   0,  13,  22,  23,   34,  50]])

# mask of all the positive values
sign = a[0] > 0

# mask of any location where the sign changed
mask = np.insert(sign[:-1] ^ sign[1:], 0, True)

# mask all the sign changes and diff
c = np.diff(a[1,mask])

输出

In [2]: c
Out[2]: array([ 13.,  21.,  16.])