Python, Matplotlib, subplot: Hoe stel ik het asbereik in?

Hoe kan ik het bereik van de y-as van de tweede subplot instellen op b.v. [0,1000] ? De FFT plot van mijn gegevens (een kolom in een tekstbestand) resulteert in een (inf.?) spike zodat de eigenlijke gegevens niet zichtbaar zijn.

pylab.ylim([0,1000])

heeft geen effect, helaas. Dit is het hele 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()

Andere verbeteringen worden ook op prijs gesteld!

Zoals te vinden op http://www.mofeel.net/582-comp-soft-sys-matlab/54166.aspx

 pylab.ylim([0,1000])

Opmerking: Het commando moet na de plot worden uitgevoerd!

Commentaren (4)

Het gebruik van as-objecten is hiervoor een goede aanpak. Het helpt als je met meerdere figuren en subplots wilt werken. Om de as-objecten rechtstreeks toe te voegen en te manipuleren:

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

Soms wilt u de grenzen van de assen instellen voordat u de gegevens plot. In dat geval kun je de "autoscaling" functie van het Axes of AxesSubplot object instellen. De functies van belang zijn set_autoscale_on, set_autoscalex_on, en set_autoscaley_on.

In uw geval wilt u de grenzen van de y-as bevriezen, maar de x-as toestaan om uit te breiden om uw gegevens aan te passen. Daarom moet u de autoscaley_on eigenschap veranderen in False. Hier is een aangepaste versie van de FFT subplot fragment uit uw code:

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