我怎样才能将第二个子图的y轴范围设置为例如[0,1000]? 我的数据的FFT图(文本文件中的一列)导致了一个(inf.?)尖峰,所以实际数据是不可见的。
pylab.ylim([0,1000])
没有效果,很遗憾。这就是整个脚本。
# based on http://www.swharden.com/blog/2009-01-21-signal-filtering-with-python/
import numpy, scipy, pylab, random
xs = []
rawsignal = []
with open("test.dat", 'r') as f:
for line in f:
if line[0] != '#' and len(line) > 0:
xs.append( int( line.split()[0] ) )
rawsignal.append( int( line.split()[1] ) )
h, w = 3, 1
pylab.figure(figsize=(12,9))
pylab.subplots_adjust(hspace=.7)
pylab.subplot(h,w,1)
pylab.title("Signal")
pylab.plot(xs,rawsignal)
pylab.subplot(h,w,2)
pylab.title("FFT")
fft = scipy.fft(rawsignal)
#~ pylab.axis([None,None,0,1000])
pylab.ylim([0,1000])
pylab.plot(abs(fft))
pylab.savefig("SIG.png",dpi=200)
pylab.show()
也希望能有其他改进!
使用轴对象是一个很好的方法。如果你想与多个数字和子图互动,它是有帮助的。要直接添加和操作坐标轴对象。
import matplotlib.pyplot as plt
fig = plt.figure(figsize=(12,9))
signal_axes = fig.add_subplot(211)
signal_axes.plot(xs,rawsignal)
fft_axes = fig.add_subplot(212)
fft_axes.set_title("FFT")
fft_axes.set_autoscaley_on(False)
fft_axes.set_ylim([0,1000])
fft = scipy.fft(rawsignal)
fft_axes.plot(abs(fft))
plt.show()
有时你真的想在绘制数据之前*设置轴的限制。 在这种情况下,你可以设置Axes
或AxesSubplot
对象的"自动缩放" 功能。 有关的函数是set_autoscale_on
,set_autoscalex_on
和set_autoscaley_on
。
在你的案例中,你想冻结y轴'的限制,但允许x轴扩展以适应你的数据。 因此,你想把autoscaley_on
属性改为False
。 下面是你代码中FFT子图片段的修改版本。
fft_axes = pylab.subplot(h,w,2)
pylab.title("FFT")
fft = scipy.fft(rawsignal)
pylab.ylim([0,1000])
fft_axes.set_autoscaley_on(False)
pylab.plot(abs(fft))