Python，Matplotlib，子图：如何设置轴范围？

Question

如何将第二个子图的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()

其他改进也表示赞赏！

Answer 1

有时，您确实想在绘制数据之前设置轴限制。在这种情况下，您可以设置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))