有效检测Python中的符号变化

我想做这个人所做的事情：

Python - 计算符号变化

但是我需要优化它以使其运行速度超快。简而言之，我想获取一个时间序列并告诉它每次穿过零（改变符号）。我想记录过零之间的时间。由于这是真实数据（32 位浮点数），我怀疑每个数字都会为零，所以这并不重要。我目前有一个计时程序，所以我会为你的结果计时，看看谁赢了。

我的解决方案给出（微秒）：

open data       8384
sign data       8123
zcd data        415466

正如您所看到的，过零检测器是缓慢的部分。这是我的代码。

import numpy, datetime

class timer():
    def __init__(self):
        self.t0 = datetime.datetime.now()
        self.t = datetime.datetime.now()
    def __call__(self,text='unknown'):
        print text,'\t',(datetime.datetime.now()-self.t).microseconds
        self.t=datetime.datetime.now()

def zcd(data,t):
    sign_array=numpy.sign(data)
    t('sign data')
    out=[]
    current = sign_array[0]
    count=0
    for i in sign_array[1:]:
        if i!=current:
            out.append(count)
            current=i
            count=0
        else: count+=1
    t('zcd data')
    return out

def main():
    t = timer()
    data = numpy.fromfile('deci.dat',dtype=numpy.float32)
    t('open data')
    zcd(data,t)

if __name__=='__main__':
    main()

关于什么：

import numpy
a = [1, 2, 1, 1, -3, -4, 7, 8, 9, 10, -2, 1, -3, 5, 6, 7, -10]
zero_crossings = numpy.where(numpy.diff(numpy.sign(a)))[0]

Output:

> zero_crossings
array([ 3,  5,  9, 10, 11, 12, 15])

即，zero_crossings 将包含元素的索引before发生过零。如果你想要元素after，只需向该数组添加 1 即可。