在python中使用heapq函数‘heapq’查找FFT的峰值及其对应的频率
在python中使用heapq函数‘heapq’查找FFT的峰值及其对应的频率
提问于 2013-06-04 01:36:37
我正在使用FFT来查看我正在设计的IC测试仪的输出信号上的失真。我有两个数组，一个包含采样频率，另一个包含相应的FFT值。我已经能够使用heapq库中的nlargest函数打印出三个最高的FFT值，但我还想打印x轴数组中相应的频率值。我的部分代码如下。频率值存储在数组'frq‘中，而FFT值存储在数组'Y’中。
Y = sci.fft(y)/n # fft computing and normalization
Y = Y[range(n/2)]
Y = abs(Y)
print heapq.nlargest(3, 20*np.log10(abs(Y)))
print heapq.nlargest(3, frq, key=lambda i: Y[i])我从最后一行代码中收到以下错误：
print heapq.nlargest(2, frq, key=lambda i: Y[i])
IndexError: index 500 is out of bounds for axis 0 with size 50回答 2
Stack Overflow用户
回答已采纳
发布于 2013-06-04 01:51:31
下面是一种方法，它将给出3个对应于最高元组的元组
from itertools import izip
print heapq.nlargest(3, izip(frq, 20*np.log10(abs(Y))), key=lambda x: x[1])您也可以将Y放在第一位
print heapq.nlargest(3, izip(20*np.log10(abs(Y)), frq))Stack Overflow用户
发布于 2013-06-04 03:44:23
numpythonic这样做的方法是完全跳过heapq，像这样：
idx = np.argsort(Y)[::-1][:3]
y_top_3 = 20*np.log10(Y[idx])
f_top_3 = frq[idx]如果你想得到一个可迭代的迭代器，你可以这样做：
top_3 = np.vstack(f_top_3, y_top_3).T页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/16902557
复制相关文章
相似问题
腾讯云开发者
