如何在数组中找到两个连续的正负值?

时间:2016-02-08 20:06:34

标签: python arrays algorithm sign

我有以下数组:

X
array([ 3.5, -3, 5.4, 3.7, 14.9, -7.8, -3.5, 2.1])

对于X的每个值,我知道它的录制时间T。我想找到两个连续正负或反之间的指数。总结我想要一个像

这样的数组
Y = array([ T(1)-T(0), T(2)-T(1),  T(5)-T(4), T(7)-T(6)])

1 个答案:

答案 0 :(得分:2)

也许迭代列表中的数组理解对你有用:

In [35]: x=np.array([ 3.5, -3, 5.4, 3.7, 14.9, -7.8, -3.5, 2.1])

In [36]: y=np.array([b-a for a,b in zip(x, x[1:]) if (a<0) != (b<0)])

In [37]: y
Out[37]: array([ -6.5,   8.4, -22.7,   5.6])

修改

我显然完全不理解这个问题。试试这个:

In [38]: X=np.array([ 3.5, -3, 5.4, 3.7, 14.9, -7.8, -3.5, 2.1])

In [39]: T=np.array([ 0, 0.1, 2, 3.5, 5, 22, 25, 50])

In [40]: y=np.array([t1-t0 for x0,x1,t0,t1 in zip(X, X[1:], T, T[1:]) if (x0<0) != (x1<0)])

In [41]: y
Out[41]: array([  0.1,   1.9,  17. ,  25. ])