Skip to content
This repository was archived by the owner on Aug 2, 2022. It is now read-only.

Commit d6a1134

Browse files
committed
Fixed a bug when correcting r-peaks (closes #35).
1 parent 2b749a5 commit d6a1134

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

biosppy/signals/ecg.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -439,6 +439,7 @@ def correct_rpeaks(signal=None, rpeaks=None, sampling_rate=1000., tol=0.05):
439439
break
440440
newR.append(a + np.argmax(signal[a:b]))
441441

442+
newR = sorted(list(set(newR)))
442443
newR = np.array(newR, dtype='int')
443444

444445
return utils.ReturnTuple((newR,), ('rpeaks',))
@@ -885,6 +886,7 @@ def gamboa_segmenter(signal=None, sampling_rate=1000., tol=0.002):
885886
previous = i
886887
rpeaks.append(np.argmax(signal[int(i):int(i + v_100ms)]) + i)
887888

889+
rpeaks = sorted(list(set(rpeaks)))
888890
rpeaks = np.array(rpeaks, dtype='int')
889891

890892
return utils.ReturnTuple((rpeaks,), ('rpeaks',))

0 commit comments

Comments
 (0)