Python, Matplotlib, subplot : comment définir l'étendue de l'axe ?

Comment puis-je définir la plage de l'axe des y du deuxième sous-graphe sur [0,1000], par exemple ? Le tracé FFT de mes données (une colonne dans un fichier texte) résulte en un pic (inf. ?) de sorte que les données réelles ne sont pas visibles.

pylab.ylim([0,1000])

n'a aucun effet, malheureusement. Voici le script complet :

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

D'autres améliorations sont également appréciées !

Comme on le trouve sur http://www.mofeel.net/582-comp-soft-sys-matlab/54166.aspx

 pylab.ylim([0,1000])

Note : La commande doit être exécutée après le tracé !

Commentaires (4)

L'utilisation des [objets axes][1] est une excellente approche pour cela. Elle est utile si vous souhaitez interagir avec plusieurs figures et sous-plans. Pour ajouter et manipuler les objets axes directement :

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

[1] : https://matplotlib.org/api/axes_api.html

Commentaires (1)

Parfois, vous voulez vraiment définir les limites des axes avant de tracer les données. Dans ce cas, vous pouvez définir la fonction "autoscaling&quot ; de l'objet Axes ou AxesSubplot. Les fonctions intéressantes sont set_autoscale_on, set_autoscalex_on, et set_autoscaley_on.

Dans votre cas, vous voulez geler les limites de l'axe des y, mais permettre à l'axe des x de s'étendre pour s'adapter à vos données. Par conséquent, vous voulez changer la propriété autoscaley_on en False. Voici une version modifiée de l'extrait du sous-graphe FFT de votre 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))
Commentaires (0)