Matplotlib에서 컬러바 위치를 오른쪽으로 이동하는 방법

Matplotlib에서 그래프의 컬러바(colorbar) 위치를 오른쪽으로 이동하려면 pad 매개변수를 활용하면 됩니다. 아래에서 단계별로 자세히 알아보겠습니다.

구현 단계

• numpy와 matplotlib 라이브러리를 임포트합니다.
• 그림 크기를 설정하고 서브플롯 간의 여백을 조정합니다.
• 샘플 데이터의 개수를 저장할 변수 N을 초기화합니다.
• numpy를 사용하여 x, y 데이터 포인트를 생성합니다.
• scatter() 메서드를 사용해 산점도(scatter plot)를 그립니다.
• colorbar() 메서드로 플롯에 컬러바를 추가하고, pad 값을 조절하여 컬러바를 좌우로 이동시킵니다.
• show() 메서드를 사용하여 그림을 화면에 표시합니다.

예제 코드

# numpy와 matplotlib 임포트
import numpy as np
from matplotlib import pyplot as plt

# 그림 크기 설정
plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

N = 100

# x, y 데이터 포인트 생성
x = np.random.rand(N)
y = np.random.rand(N)

# x, y 데이터로 산점도 그리기
s = plt.scatter(x, y, c=x, cmap='hot', marker='*')

# pad 값을 지정하여 컬러바 추가
plt.colorbar(s, shrink=0.9, pad=0.1)

# 플롯 표시
plt.show()

실행 결과

위 코드를 실행하면 다음과 같은 출력 결과를 얻을 수 있습니다.

컬러바를 왼쪽에 배치하는 방법

반대로 컬러바를 그래프의 왼쪽에 배치하고 싶다면, location 매개변수를 사용하면 됩니다.

plt.colorbar(s, shrink=0.9, pad=0.1, location="left")

이 코드를 적용하면 컬러바가 그래프 왼쪽에 배치된 것을 확인할 수 있습니다.

정리

Matplotlib에서 컬러바의 위치를 조정하는 핵심은 두 가지입니다. 첫째, pad 값으로 컬러바와 그래프 사이의 간격을 좌우로 조절할 수 있고, 둘째, location 매개변수에 "left", "right", "top", "bottom" 등의 값을 지정하여 컬러바의 배치 방향을 자유롭게 설정할 수 있습니다. 이 두 매개변수를 적절히 활용하면 시각화 결과물의 가독성을 크게 높일 수 있습니다.