Python, Matplotlib, subplot: Como definir o intervalo do eixo?

Como posso definir o intervalo do eixo y da segunda subtrama para, por exemplo, [0,1000]? O gráfico FFT dos meus dados (uma coluna em um arquivo de texto) resulta em um pico (inf.?) Para que os dados reais não sejam visíveis.

pylab.ylim([0,1000])

infelizmente não tem efeito. Este é o script inteiro:

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

Outras melhorias também são apreciadas!

Conforme encontrado em http://www.mofeel.net/582-comp-soft-sys-matlab/54166.aspx

 pylab.ylim([0,1000])

Nota: O comando deve ser executado após a plotagem!

Usar objetos de eixos é uma ótima abordagem para isso. Ajuda se você deseja interagir com várias figuras e subparcelas. Para adicionar e manipular os objetos de eixos diretamente:

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

Às vezes, você realmente deseja definir os limites dos eixos antes de plotar os dados. Nesse caso, você pode definir o recurso "escalonamento automático" do objeto Axesou AxesSubplot. As funções de interesse são set_autoscale_on, set_autoscalex_one set_autoscaley_on.

No seu caso, você deseja congelar os limites do eixo y, mas permita que o eixo x se expanda para acomodar seus dados. Portanto, você deseja alterar a autoscaley_onpropriedade para False. Aqui está uma versão modificada do snippet de subtrama FFT do seu código:

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

Se você sabe o eixo exato que deseja, então

pylab.ylim([0,1000])

funciona como respondido anteriormente. Mas se você deseja que um eixo mais flexível se ajuste aos dados exatos, como fiz quando encontrei essa pergunta, defina o limite do eixo como o comprimento do seu conjunto de dados. Se o seu conjunto de dados for fftcomo na pergunta, adicione-o após o comando plot:

length = (len(fft)) pylab.ylim([0,length])

Se você tiver várias subparcelas, ou seja,

fig, ax = plt.subplots(4, 2)

Você pode usar os mesmos y limites para todos eles. Ele obtém limites de y ax do primeiro gráfico.

plt.setp(ax, ylim=ax[0,0].get_ylim())