1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
import matplotlib.pyplot as plt
def plot(x, y, ind):
"""
Plots the original data with the peaks that were identified
Parameters
----------
x : array-like
Data on the x-axis
y : array-like
Data on the y-axis
ind : array-like
Indexes of the identified peaks
"""
plt.plot(x, y, "--")
marker_x = x.iloc[ind] if hasattr(x, "iloc") else x[ind]
marker_y = y.iloc[ind] if hasattr(y, "iloc") else y[ind]
plt.plot(marker_x, marker_y, "r+", ms=5, mew=2, label="{} peaks".format(len(ind)))
plt.legend()
|
