Python, Matplotlib, subplot : 축 범위를 설정하는 방법?


266

두 번째 서브 플롯의 y 축 범위를 어떻게 [0,1000]으로 설정합니까? 내 데이터 (텍스트 파일의 열)의 FFT 플롯은 실제 데이터가 표시되지 않도록 (inf ??) 스파이크를 발생시킵니다.

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 December

1
내가 hexbin과 함께 이것을 사용한다면, plot ()이 두 음모 모두에서 흰색 배경을 노출 한 후 ylim을 사용합니다
lynxoid

3
플롯을 사용하지 않고 savefig ??
Ben

8
전화 plot()한 후, ylim()다음 savefig().
therealrootuser 2016

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
Rob이 제안한 것처럼 matplotlib의 OO 인터페이스는 상태 기반 pylab 인터페이스보다 선호됩니다. "많은 예제는 pylab을 사용하지만 더 이상 권장되지 않습니다. 비 대화식 플로팅의 경우 pyplot을 사용하여 그림을 만든 다음 플로팅을위한 OO 인터페이스를 사용하는 것이 좋습니다." matplotlib.org/faq/…
베넷 브라운

26

때로는 데이터를 플롯 하기 전에 축 제한을 설정하려고 할 때가 있습니다 . 이 경우 Axes또는 AxesSubplot개체 의 "자동 크기 조정"기능을 설정할 수 있습니다 . 관심의 기능은 set_autoscale_on, set_autoscalex_on하고 set_autoscaley_on.

귀하의 경우 y 축의 한계를 고정하고 싶지만 x 축이 데이터를 수용하도록 확장되도록하십시오. 따라서 autoscaley_on속성을 로 변경하려고 합니다 False. 다음은 코드에서 수정 된 FFT 서브 플롯 스 니펫 버전입니다.

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())
당사 사이트를 사용함과 동시에 당사의 쿠키 정책개인정보 보호정책을 읽고 이해하였음을 인정하는 것으로 간주합니다.
Licensed under cc by-sa 3.0 with attribution required.