如何设置子图轴范围

Question

我该如何设置第二个子图的y轴范围，比如说设置为[0,1000]呢？我对数据做了快速傅里叶变换（FFT），结果出现了一个很大的尖峰，导致实际的数据看不见。

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))

Answer 2

使用坐标轴对象是一个很好的方法。这种方法特别适合当你想同时处理多个图形和子图时。要直接添加和操作坐标轴对象，可以这样做：

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

Answer 3

你可以使用 pylab.ylim 来设置图表的y轴范围。

pylab.ylim([0,1000])

注意：这个命令必须在绘制图表之后执行！

更新 2021
因为 matplotlib 现在强烈不推荐使用 pylab，所以你应该改用 pyplot：

from matplotlib import pyplot as plt
plt.ylim(0, 100) 
#corresponding function for the x-axis
plt.xlim(1, 1000)