Python, Matplotlib, subplot: Как установить диапазон осей?

Как установить диапазон оси y второго подзаголовка, например, [0,1000]? График FFT моих данных (столбец в текстовом файле) приводит к всплеску (inf.?), Так что фактические данные не видны.

pylab.ylim([0,1000])

не действует, к сожалению. Это целый script:

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

Другие улучшения также приветствуются!

ОТВЕТЫ

Ответ 2

Использование объектов axes - отличный подход для этого. Это помогает, если вы хотите взаимодействовать с несколькими фигурами и подзаголовками. Для непосредственного добавления и управления объектами осей:

fig = pylab.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))

pylab.show()

Ответ 3

Иногда вы действительно хотите установить лимиты осей перед тем, как вы построите данные. В этом случае вы можете установить функцию "автомасштабирование" объекта 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))