我如何在Python中找到峰？我得到索引。我想要价值

Question

from scipy.signal import find_peaks
a=[2,3,4,1]
b=find_peaks(a)
print(b)

我得到了索引。但是我想要价值观。如何获得列表的值？

Answer 1

您可以使用原始矢量上的索引来返回峰值的值：

a = np.array([2, 3, 4, 1])
peaks, properties = find_peaks(a)
a[peaks]

Answer 2

最简单的方法是将a转换为numpy数组，然后将其索引。

a = np.array([2, 3, 4, 1, 5, 0], dtype=np.float32)
idx, props = find_peaks(a)
val = a[idx]

print(idx)  # this outputs [2 4]
print(val)  # this outputs [4. 5.]