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

Question

我有以下数组：

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)])

Answer 1

也许在列表理解中迭代数组对你有用：

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. ])

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

How to find two consecutive positive-negative values in an array?

python

arrays

algorithm

sign

编辑