Update ecg_segment.py

modify plotting error
neuropsychology · Oct 28, 2023 · 4d4bd7a · 4d4bd7a
1 parent ce33299
commit 4d4bd7a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neurokit2/ecg/ecg_segment.py b/neurokit2/ecg/ecg_segment.py
@@ -112,7 +112,7 @@ def _ecg_segment_plot(heartbeats, heartrate=0, ytitle="ECG", color="#F44336", ax
     # Plot average heartbeat
     ax.plot(
         mean_heartbeat.index,
-        mean_heartbeat,
+        mean_heartbeat[col].values,
         color=color,
         linewidth=7,
         label="Average beat shape",