그래프 사이즈 조절하기

import numpy as np
import matplotlib.pyplot as plt

height_array = np.array([
    165, 164, 155, 151, 157, 162, 155, 157, 165, 162,
    165, 167, 167, 183, 180, 184, 177, 178, 175, 181,
    172, 173, 169, 172, 177, 178, 185, 186, 190, 187
])

weight_array = np.array([
    62, 59, 57, 55, 60, 58, 51, 56, 68, 64,
    57, 58, 64, 79, 73, 76, 61, 65, 83, 80,
    67, 82, 88, 62, 61, 79, 81, 68, 83, 80
])

산점도를 그려보자. 그래프의 크기를 기본적으로 인치 단위로 표현한다. 따로 설정하지 않으면 가로 6인치, 세로 4인치 크기 그래프가 나온다.

plt.scatter(height_array, weight_array)
plt.title('Height and Weight')
plt.xlabel('Height (cm)')
plt.ylabel('Weight (kg)')
plt.show()

기본값 그래프

---

다른 크기의 그래프를 원할 때에는 pyplot의 figure() 함수를 호출해서 그래프의 크기를 변경하면 된다. 이 방법은 그래프 하나하나의 크기를 설정할 때 사용한다. 이번에만 적용되고 다음 그래프 그릴 때에는 다시 기본값 (가로 6인치, 세로 4인치)로 초기화 된다.

plt.figure(figsize=(10, 4)) 
plt.scatter(height_array, weight_array)
plt.title('Height and Weight')
plt.xlabel('Height (cm)')
plt.ylabel('Weight (kg)')
plt.show()

가로 10인치 세로 4인치

---

전체적으로 그래프 크기를 통일하고 싶은 경우 pyplot 의 rcParams 속성에서 직접 접근해서 그래프 사이즈의 기본 설정을 변경할 수 있다.

앞으로 같은 노트북 파일 안에서 그래프를 그리면 따로 설명을 바꿔주지 않는 한 계속 가로 5인치, 세로 5인치 크기의 그래프가 나온다.

plt.rcParams['figure.figsize'] = (5, 5)
plt.scatter(height_array, weight_array)
plt.title('Scatter Plot')
plt.xlabel('X Label')
plt.ylabel('Y Label')
plt.show()

가로 5인치 세로 5인