Python, Matplotlib, subplot: Як встановити діапазон осей?


266

Як я можу встановити діапазон осей y другої підплоти наприклад [0,1000]? Діаграма FFT моїх даних (стовпець у текстовому файлі) призводить до (інф.?) Шипу, так що фактичні дані не видно.

pylab.ylim([0,1000])

не має ефекту, на жаль. Це весь сценарій:

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

Також вдячні інші вдосконалення!


Відповіді:


284

Як знайдено в http://www.mofeel.net/582-comp-soft-sys-matlab/54166.aspx

 pylab.ylim([0,1000])

Примітка: Команда повинна бути виконана після сюжету!


3
Щоразу, коли я це роблю, він перевертає зображення догори дном.
ely

1
якщо я використовую це з hexbin, використовує ylim після plot () виставляє білий фон на обох ділянках
lynxoid

3
Що, якщо ви не використовуєте сюжет, а savefig ??
Бен

8
Телефонуйте plot(), то ylim()і тоді savefig().
therealrootuser

112

Використання осей-об’єктів - чудовий підхід для цього. Це допомагає, якщо ви хочете взаємодіяти з декількома фігурами та піддіаграмами. Для додавання та маніпулювання об'єктами осей безпосередньо:

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

2
Як пропонує Роб, інтерфейс OO в matplotlib є кращим перед інтерфейсом пілабу на основі стану. "Хоча в багатьох прикладах використовується pylab, він більше не рекомендується. Для інтерактивного побудови графіків пропонується використовувати pyplot для створення фігур, а потім інтерфейс OO для побудови графіків." matplotlib.org/faq/…
Беннет Браун

26

Іноді дуже хочеться встановити межі осей, перш ніж будувати дані. У цьому випадку ви можете встановити функцію "автоматичного масштабування" об'єкта Axesабо AxesSubplotоб'єкта. Функції інтерес set_autoscale_on, set_autoscalex_onі set_autoscaley_on.

У вашому випадку ви хочете заморозити межі осі y, але дозволити розширення осі x для розміщення ваших даних. Тому ви хочете змінити autoscaley_onвластивість на False. Ось модифікована версія фрагмента FFT subplot з вашого коду:

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

0

Якщо ви знаєте точну вісь, яку ви хочете, то

pylab.ylim([0,1000])

працює, як відповіли раніше. Але якщо ви хочете, щоб більш гнучка вісь відповідала вашим точним даним, як я це робив, коли я знайшов це питання, тоді встановіть межа осі довжиною вашого набору даних. Якщо ваш набір даних fftяк у запитанні, додайте це після команди своєї графіки:

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


0

Якщо у вас є кілька субплот, тобто

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

Ви можете використовувати однакові y обмеження для всіх. Він отримує межі y ax від першого сюжету.

plt.setp(ax, ylim=ax[0,0].get_ylim())
Використовуючи наш веб-сайт, ви визнаєте, що прочитали та зрозуміли наші Політику щодо файлів cookie та Політику конфіденційності.
Licensed under cc by-sa 3.0 with attribution required.