`find_local_peaks` ignores the sample at `i + radius`, so points on a rising slope are returned as peaks
还没有人认领这个 Issue。
评估
调研方向
从 wfdb/processing/peaks.py 中的 find_local_peaks 开始,检查其循环中的窗口切片。使用提供的复现来验证每个半径都会考虑右侧边界,然后检查 qrs.py 中 XQRS 调用点的兼容性。在不更改单独的 flat-top TODO 的情况下,输出与参考用例匹配即表示完成。
由索引模型根据 Issue 内容生成。
描述
The docstring of wfdb.processing.find_local_peaks says a sample is a local peak "if it is the largest value within the samples on its left and right". The window it actually checks leaves out the right side:
# wfdb/processing/peaks.py
while i < radius + 1:
if sig[i] == max(sig[: i + radius]): # right side stops at i + radius - 1
...
while i < len(sig):
if sig[i] == max(sig[i - radius : i + radius]): # sample i + radius is never looked at
Python slices exclude the end, so sig[i + radius] is never compared. With radius=1 the right neighbour is not checked at all, so every sample on a rising edge counts as a peak. With bigger radii a sample still wins if the only larger value is exactly radius samples to its right.
Reproduction
import numpy as np
from wfdb.processing import find_local_peaks
def reference(sig, radius):
# what the docstring describes
out = []
for i in range(len(sig)):
lo, hi = max(0, i - radius), min(len(sig), i + radius + 1)
if sig[i] == sig[lo:hi].max():
out.append(i)
return out
for sig, r in [
(np.array([0, 1, 2, 3, 2, 1, 0], dtype=float), 1),
(np.array([0, 1, 3, 2, 10, 5, 1, 0], dtype=float), 2),
]:
print(sig.astype(int).tolist(), "radius", r)
print(" find_local_peaks:", find_local_peaks(sig, r).tolist())
print(" expected :", reference(sig, r))
Output:
[0, 1, 2, 3, 2, 1, 0] radius 1
find_local_peaks: [0, 1, 2, 3]
expected : [3]
[0, 1, 3, 2, 10, 5, 1, 0] radius 2
find_local_peaks: [2, 4]
expected : [4]
In the first case indices 0, 1 and 2 are all on the way up to the real maximum at index 3. In the second case index 2 (value 3) is reported even though index 4 (value 10) is only 2 samples away.
Why it matters
XQRS uses this function to pick QRS candidates (qrs.py, lines 236 and 283), so a candidate can end up on the slope next to the real peak, not on the peak itself. With real ECG sampling rates the radius is large, so this probably shows up rarely there, but anyone calling find_local_peaks directly with a small radius gets wrong results.
Possible fix
Include the right edge in both slices (sig[: i + radius + 1] and sig[i - radius : i + radius + 1]). The third loop is unreachable (already reported in #474), so it can go away in the same change. I have not touched the flat-top case that the TODO comment mentions; that looks like a separate question.
Versions
- wfdb 4.3.1 from PyPI, and also current
main(f627b5f), same output - numpy 2.5.3, Python 3.12.13, macOS 26.4.1
- 主要语言
- Jupyter Notebook
- 星标
- 853
- 派生
- 322
- PR 合并指标
- 30 天内没有已合并 PR
环境准备
这个项目没有提供开发容器、Dockerfile 或贡献指南,环境需要你自己搭建:先看它的 README,通用步骤见我们的新手贡献指南。
从这里开始
- 先读完整个 Issue,再读项目的贡献指南。
- 在 Issue 下留言说明你要接手 —— 这能避免两个人做同样的事。
- Fork 仓库,在一个分支上完成修改。
- 提交 Pull Request,并在描述里引用这个 Issue 编号。
MIT-LCP/wfdb-python 的其他 Issue
-
难度 4/5 3-5 天 新手友好度 45/100
MIT-LCP/wfdb-python#568 ·
-
难度 3/5 1-2 天 新手友好度 48/100
MIT-LCP/wfdb-python#557 ·
-
难度 2/5 1-3 小时 新手友好度 58/100
MIT-LCP/wfdb-python#554 ·
-
难度 4/5 3-5 天 新手友好度 35/100
MIT-LCP/wfdb-python#545 ·
-
难度 5/5 一周以上 新手友好度 30/100
MIT-LCP/wfdb-python#540 ·
查看 MIT-LCP/wfdb-python 的全部 Issue
相似的 Issue
-
难度 2/5 1-3 小时 新手友好度 74/100
维护者通常 1 天内回复
-
难度 2/5 1-3 小时 新手友好度 72/100
BSData/age-of-sigmar-4th#1378 ·
-
难度 1/5 1 小时以内 新手友好度 90/100
semanticpixel/theluistorres#224 ·
维护者通常 1 天内回复
-
难度 1/5 1 小时以内 新手友好度 88/100
donnemartin/system-design-primer#1407 ·
维护者通常 1 天内回复
-
难度 2/5 1-3 小时 新手友好度 88/100
维护者通常 7 天内回复