python - 修剪 matplotlib 功率谱图

Question

我的第一个stackoverflow问题，希望我做对了..

我正在尝试构建功率谱密度图 ( matplotlib.pyplot.psd)，并且我需要在绘制之前丢弃频谱两端的 256 个 bin。对于我的任务，我需要实际删除数据点，而不是简单地限制 x 轴范围。

psd 函数返回两个数组的元组，我想我可以简单地删除数组的元素，然后调用show(). 但它似乎不起作用。

from matplotlib.pyplot import *
import numpy as np

#put some dummy data into s
s=np.vectorize(complex)(range(2048),range(2048))

x=psd(s, 2048, 194171, window=np.bartlett(2048),Fc=14050000)

#trim x
del x[0][-256:]
del x[0][:256]
del x[1][-256:]
del x[1][:256]

show()

这给出了：

Traceback (most recent call last):
  File "dummy-fft.py", line 10, in <module>
    del x[0][-256:]
ValueError: cannot delete array elements

有没有更好的方法来截断 psd 图？

Answer 1

您可以使用 slice 来获取所需的部分，而不是删除头部和尾部：

from matplotlib.pyplot import *
import numpy as np

#put some dummy data into s
s=np.vectorize(complex)(range(2048),range(2048))

power, freq=psd(s, 2048, 194171, window=np.bartlett(2048),Fc=14050000)
figure()
semilogy(freq[256:-256], power[256:-256])
show()