首页 文章

Python,Matplotlib,subplot:如何设置轴范围?

提问于
浏览
240

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

其他改进也值得赞赏!

3 回答

  • 264

    使用axes objects是一个很好的方法 . 如果您想要与多个图形和子图形进行交互,它会有所帮助 . 直接添加和操作轴对象:

    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()
    
  • 105

    见于http://www.mofeel.net/582-comp-soft-sys-matlab/54166.aspx

    pylab.ylim([0,1000])
    

    注意:命令必须在绘图后执行!

  • 26

    有时您真的想在绘制数据之前设置轴限制 . 在这种情况下,您可以设置 AxesAxesSubplot 对象的"autoscaling"功能 . 感兴趣的功能是 set_autoscale_onset_autoscalex_onset_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))
    

相关问题