Python检测数据帧中的零交叉

Question

我有一个数据框。我想检测数据帧的零交叉:

我的代码:

def zero_crossing(data):
    return np.where(np.diff(np.sign(np.array(data))))[0]

df1 = pd.DataFrame({'a':[1,2,-1,5,0,9,-6,7]},index=[100,101,102,103,104,105,106,107])

df1 = 
     a
100  1
101  2
102 -1
103  5
104  0
105  9
106 -6
107  7
print(zero_crossing(df1['a']))

当前输出:

array([1, 2, 3, 4, 5, 6], dtype=int64)

预期输出:

[101,102,103,104,105,106]

Answer 1

使用你的结果

df1.index[zero_crossing(df1['a'])]
Out[132]: Int64Index([101, 102, 103, 104, 105, 106], dtype='int64')

#df1.index[zero_crossing(df1['a'])].tolist()
#Out[133]: [101, 102, 103, 104, 105, 106]