[파이썬 matplotlib] 이미지맵(imshow) 범위설정하기

[파이썬 matplotlib] 이미지맵(imshow) 범위설정하기

이미지맵의 x축과 y축 범위는 index 로 되어 있습니다. 아래는 예시입니다. 

 

import numpy as np
import matplotlib.cm as cm
from matplotlib import pyplot as plt
import scipy.stats as stats


X=np.linspace(-4,4,100)
Y=np.linspace(-4,4,100)


Z=np.zeros((100,100))  

for i,x in enumerate(X):
    for j,y in enumerate(X):
        Z[i,j]=stats.norm(0, 1).pdf(X[i])*stats.norm(0, 1).pdf(Y[j]) 

plt.imshow(Z,cm.gray)
plt.show()

 

 

index를 원하는 범위로 변경하고 싶을 경우 extent 옵션을 설정합니다. 아래와 같은 형식입니다.

extent=(xmin,xmax,ymin,ymax)

 

위 그래프에 적용해봅시다. plt.imshoq(Z,cm.gray) 부분을 아래와 같이 변경하면 됩니다. 

 

plt.imshow(Z,cm.gray,extent=(-4,4,-4,4))