问如何使用python从离散采样数据点获得最高的两个峰值？

EN

import matplotlib.pyplot as plt
import numpy as np


# My data set (example is a crazy cosine wave)
x = np.linspace(1, 100, 1000)
y = np.cos(x*0.5)*((x - 50)**3)

# Find first derivative:
m = np.diff(y)/np.diff(x)

# Get indicies of peaks and valleys
c = len(m)
indices = []
for i in range(1, c):
    prev_val = m[i-1]
    val = m[i]
    if prev_val < 0 and val >= 0:
        indices.append(i)
    elif prev_val > 0 and val <= 0:
        indices.append(i)

# Get the values, positions, and indicies of the two peaks
max_list = [0, 0]
index_list = [0, 0]
pos_list = [0, 0]
for index in indices:
    val = y[index]
    if val > max_list[0]:
        max_list[0] = val
        index_list[0] = index
        pos_list[0] = x[index]
    elif val > max_list[1]:
        max_list[1] = val
        index_list[1] = index
        pos_list[1] = x[index]


print('Two peak indices:', index_list)
print('Two peak values:', max_list)
print('Two peak x-positions:', pos_list)

average_pos = (pos_list[0] + pos_list[1])/2

print('Average x-position:', average_pos)



plt.plot(x, y)
plt.show()