개요
Matplotlib로 그린 그래프의 색상 정보를 확인하고 싶다면, 각 플롯 객체에 대해 get_color() 메서드를 사용하면 됩니다. 이 메서드는 해당 선(line)에 적용된 색상 값을 헥스 코드 형태의 문자열로 반환해 주기 때문에, 나중에 같은 색상을 다른 그래프에 재사용하거나 시각화 스타일을 일관되게 유지할 때 매우 유용합니다.
구현 단계
- 그림(figure) 크기를 설정하고, 서브플롯 주변과 사이의 여백(padding)을 조정합니다.
- NumPy를 사용하여 x와 y 데이터 포인트를 생성합니다.
- plot() 메서드를 사용하여 (x, y), (x, y²), (x, y³) 세 개의 곡선을 한 번에 그립니다.
- 각 그래프 선에 대한 범례(legend)를 배치합니다.
- get_color() 메서드를 호출하여 각 플롯의 색상을 가져옵니다.
- show() 메서드를 사용하여 그림을 화면에 표시합니다.
예제 코드
import numpy as np
import matplotlib.pyplot as plt
plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True
x = np.arange(10)
y = np.arange(10)
p = plt.plot(x, y, x, y ** 2, x, y ** 3)
plt.legend([p[0], p[1], p[2]], ["$y=x$", "$y=x^2$", "$y=x^3$"])
print("첫 번째 그래프의 색상: ", p[0].get_color())
print("두 번째 그래프의 색상: ", p[1].get_color())
print("세 번째 그래프의 색상: ", p[2].get_color())
plt.show()실행 결과
코드를 실행하면 아래와 같이 세 개의 곡선과 범례가 함께 표시된 그래프 창이 나타납니다.
동시에 콘솔에는 각 그래프의 색상 값이 다음과 같이 출력됩니다.
Color of the first plot: #1f77b4 Color of the second plot: #ff7f0e Color of the third plot: #2ca02c
참고 사항
위 출력 결과에서 볼 수 있듯이, Matplotlib의 기본 색상 순환(cycle)은 #1f77b4(파랑), #ff7f0e(주황), #2ca02c(녹색) 순서로 적용됩니다. plt.plot()은 여러 개의 선을 한 번에 그릴 경우 Line2D 객체들의 리스트를 반환하므로, 인덱스(p[0], p[1], p[2])로 각 선에 접근한 뒤 get_color()를 호출하면 해당 선의 정확한 색상 코드를 얻을 수 있습니다. 이렇게 얻은 색상 값은 color 매개변수에 그대로 전달하여 다른 그래프에서도 동일한 색상을 재현할 수 있습니다.