[파이썬] Matplotlib 유용한 팁과 트릭

Matplotlib은 파이썬에서 데이터 시각화를 위해 가장 일반적으로 사용되는 라이브러리 중 하나입니다. 그러나 Matplotlib을 처음 사용하는 경우에는 시각화를 만들기 위한 다양한 설정과 기능들이 조금 복잡하게 느껴질 수 있습니다. 여기에 몇 가지 유용한 팁과 트릭을 소개하겠습니다.

1. 그래프 스타일 설정하기

Matplotlib은 다양한 그래프 스타일을 제공합니다. 이를 사용하면 간단한 한 줄의 코드로 그래프의 전체적인 스타일을 설정할 수 있습니다.

import matplotlib.pyplot as plt

plt.style.use('ggplot')  # ggplot 스타일 사용
plt.plot(x, y)
plt.show()

다른 유용한 스타일로는 ‘seaborn’, ‘dark_background’, ‘fivethirtyeight’ 등이 있습니다. 원하는 스타일을 선택하여 그래프를 꾸밀 수 있습니다.

2. 색상 선택하기

Matplotlib은 다양한 색상 맵을 제공하며, 색상과 색상 맵을 선택하여 그래프를 더욱 멋지게 만들 수 있습니다.

import matplotlib.pyplot as plt

plt.plot(x, y, color='red')       # 특정 색상 사용
plt.plot(x, y, color='0.75')      # 0부터 1까지의 회색조 사용
plt.plot(x, y, color='#FF00FF')   # 16진수 RGB 색상 사용
plt.plot(x, y, cmap='viridis')    # 색상 맵 사용
plt.show()

‘color’ 매개변수를 사용하여 특정 색상을 선택하거나, 0부터 1까지의 회색조를 선택할 수 있습니다. 또는 ‘#’ 기호를 사용하여 16진수 RGB 색상을 선택할 수도 있습니다. ‘cmap’ 매개변수를 사용하여 색상 맵을 선택할 수도 있습니다.

3. 범례 추가하기

여러 개의 그래프를 한 번에 그릴 때, 각 그래프를 구분할 수 있는 범례를 추가하는 것이 유용합니다. Matplotlib에서는 ‘legend’ 함수를 사용하여 범례를 추가할 수 있습니다.

import matplotlib.pyplot as plt

plt.plot(x, y1, label='Line 1')
plt.plot(x, y2, label='Line 2')
plt.legend()
plt.show()

‘plot’ 함수에서 ‘label’ 매개변수를 사용하여 각 그래프에 레이블을 지정한 뒤, ‘legend’ 함수를 호출하여 범례를 추가합니다. 그래프를 표시하기 전에 ‘legend’ 함수를 호출해야 범례가 제대로 표시됩니다.

4. 축 범위 설정하기

Matplotlib에서는 ‘xlim’과 ‘ylim’ 함수를 사용하여 x축과 y축의 범위를 설정할 수 있습니다.

import matplotlib.pyplot as plt

plt.plot(x, y)
plt.xlim(0, 10)   # x축 범위 설정
plt.ylim(0, 5)    # y축 범위 설정
plt.show()

‘xlim’ 함수와 ‘ylim’ 함수를 사용하여 x축과 y축의 최소값과 최대값을 지정함으로써 축 범위를 설정할 수 있습니다.

Matplotlib에는 더 많은 유용한 팁과 트릭이 있지만, 이 기본적인 팁들은 새로운 사용자를 위한 시작점으로 도움이 될 것입니다. Matplotlib 공식 문서와 예제들을 참고하면 더 많은 내용을 배울 수 있습니다.