这段代码展示了如何在Python中计算Series数据的交叉点，特别是与特定值的交叉点。函数`cross()`用于找到上升、下降或两者交叉点的索引，并通过线性插值计算精确的交叉点。此外，提供了测试用例来验证其正确性。
为此,我最终得到了以下内容.它是一个矢量化版本,比使用循环的版本快150倍.
def cross(series, cross=0, direction='cross'):
"""
Given a Series returns all the index values where the data values equal
the 'cross' value.
Direction can be 'rising' (for rising edge), 'falling' (for only falling
edge), or 'cross' for both edges
"""
# Find if values are above or bellow yvalue crossing:
above=series.values > cross
below=np.logical_not(above)
left_shifted_above = above[1:]
left_shifted_below = below[1:]
x_crossings = []
# Find indexes on left side of crossing point
if direction == 'rising':
idxs = (left_shifted_above & below[0:-1]).nonzero()[0]
elif direction == 'falling':
idxs = (left_shifted_below &
最低0.47元/天 解锁文章
9207
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
