파이썬 matplotlib.pyplot.legend 함수 활용하기

파이썬 matplotlib.pyplot.legend 함수 활용하기: 그래프의 기준을 밝혀주는 보석

데이터 시각화의 세계에서 중요한 역할을 하는 matplotlib 라이브러리, 그중에서도 pyplot.legend 함수는 그래프의 주요 요소들을 명확히 구분해주는 역할을 합니다. 이 포스팅에서는 legend 함수의 다양한 활용법에 대해 알아보고, 어떻게 그래프를 더 이해하기 쉽게 만들 수 있는지 살펴보겠습니다.

matplotlib.pyplot.legend 함수 소개

그래프를 그릴 때 legend 함수는 각 데이터 시리즈에 대해 어떤 의미가 있는지를 표기하기 위해 사용됩니다. 이 기능은 특히 색상이나 선의 스타일이 다양한 경우, 시각적인 혼동을 줄이는데 큰 도움을 줍니다.

함수 시그니처

matplotlib.pyplot.legend(handles=None, labels=None, **kwargs)

매개변수:

• handles: 범례에 추가할 선, 점, 막대 등의 리스트입니다. 기본적으로는 현재 Axes의 선들이 자동으로 추가됩니다.
• labels: 각 핸들에 대응하는 텍스트 리스트입니다.
• **kwargs: 추가적인 설정을 위한 키워드 인자들입니다.

반환 값:

• 범례를 나타내는 Legend 객체를 반환합니다.

사용 예제

기본 예제

다음은 기본적인 legend 함수를 사용하여 간단한 그래프에 범례를 추가하는 예제입니다.

import matplotlib.pyplot as plt

# 데이터 생성
x = [1, 2, 3, 4]
y1 = [10, 20, 30, 40]
y2 = [40, 30, 20, 10]

# 그래프 그리기
plt.plot(x, y1, label='Series 1', color='blue', marker='o')
plt.plot(x, y2, label='Series 2', color='red', marker='s')

# 범례 추가
plt.legend()

# 제목 및 축 라벨
plt.title('Example Plot with Legend')
plt.xlabel('X-axis')
plt.ylabel('Y-axis')

# 그래프 보여주기
plt.show()

고급 예제: 위치 및 스타일 조정하기

여기서는 범례의 위치와 스타일을 조정하여 그래프의 시각적 효과를 높이는 방법을 보여줍니다.

import matplotlib.pyplot as plt

# 데이터 생성
x = [1, 2, 3, 4]
y1 = [10, 20, 30, 40]
y2 = [40, 30, 20, 10]

# 그래프 그리기
plt.plot(x, y1, label='Series 1', color='blue', linestyle='--', linewidth=2)
plt.plot(x, y2, label='Series 2', color='red', linestyle=':', linewidth=2)

# 범례 추가 및 스타일 조정
plt.legend(loc='upper left', fontsize='large', shadow=True)

# 제목 및 축 라벨
plt.title('Advanced Plot with Legend Styling')
plt.xlabel('X-axis')
plt.ylabel('Y-axis')

# 그래프 보여주기
plt.show()

결론

matplotlib.pyplot.legend 함수는 그래프의 가독성을 높여줍니다. 여러 데이터 시리즈를 포함한 그래프에서 각 시리즈의 의미를 명확히 전달할 수 있는 방법을 제공합니다. 범례를 통해 데이터의 내용을 쉽게 이해할 수 있으며, 이를 통해 데이터 분석의 효과를 극대화할 수 있습니다.

• 상황에 맞는 범례를 추가하여 그래프의 맥락을 강조해보세요!
• 지금 바로 pyplot.legend 함수를 활용하여 그래프의 품질을 더욱 높여보세요!