Definindo o intervalo do eixo da subtrama

Plano de fundo

Ao trabalhar com várias subtramas em uma visualização, torna-se necessário controlar o intervalo do eixo de cada subtrama individual para garantir a representação adequada dos dados. Esta questão explora como definir o intervalo do eixo y de uma segunda subtrama em um layout de duas subtramas. O problema surge quando um gráfico FFT exibe um pico atípico, tornando os dados desejados invisíveis.

Solução

Para resolver esse problema, use pylab.ylim([bottom, top]) após o gráfico foi criado. Os argumentos inferior e superior definem os limites inferior e superior do intervalo do eixo, respectivamente.

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.plot(abs(fft))
pylab.ylim([0,1000]) # Set the y-axis range

pylab.savefig("SIG.png",dpi=200)
pylab.show()

Melhoria

1. Migrar do Pylab para o pyplot do Matplotlib

A partir de 2021, o Matplotlib desencoraja fortemente o uso do pylab. Em vez disso, é recomendado importar pyplot especificamente:

from matplotlib import pyplot as plt

2. Use plt.ylim() em vez de pylab.ylim()

A sintaxe correta para definir o intervalo do eixo y usando pyplot é plt.ylim(). Seu uso é semelhante a pylab.ylim().

plt.ylim(0, 100) 

3. Definir valor mínimo do eixo X

Além de ajustar o intervalo do eixo y, considere definir o valor mínimo do eixo x para garantir que todo o intervalo do gráfico FFT esteja visível.

plt.xlim(1, 1000)