Настройка диапазона оси подграфика

Фон

При работе с несколькими подграфиками в визуализации возникает необходимость контролировать диапазон осей каждого отдельного подграфика для обеспечения правильного представления данных. В этом вопросе рассматривается, как установить диапазон оси Y второго подграфика в макете с двумя подграфиками. Проблема возникает, когда на графике БПФ обнаруживается выброс, делающий нужные данные невидимыми.

Решение

Чтобы решить эту проблему, используйте pylab.ylim([bottom, top]) после графика был создан. Нижний и верхний аргументы определяют нижнюю и верхнюю границы диапазона оси соответственно.

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

Улучшение

1. Переход с Pylab на pyplot Matplotlib

С 2021 года Matplotlib категорически не рекомендует использовать pylab. Вместо этого рекомендуется специально импортировать pyplot:

from matplotlib import pyplot as plt

2. Используйте plt.ylim() вместо pylab.ylim()

Правильный синтаксис для установки диапазона оси Y с помощью pyplot — plt.ylim(). Его использование аналогично pylab.ylim().

plt.ylim(0, 100) 

3. Установите минимальное значение оси X

Помимо настройки диапазона оси Y, рассмотрите возможность установки минимального значения оси X, чтобы обеспечить видимость всего диапазона графика БПФ.

plt.xlim(1, 1000)