Matplotlib 등고선 플롯에 축선 또는 원점 그리기

Matplotlib에서 등고선 플롯(contour plot)에 축선이나 원점을 표시하려면 contourf() 함수와 함께 axhline()(y=0 수평선), axvline()(x=0 수직선) 메서드를 활용할 수 있습니다.

구현 단계

• NumPy를 사용하여 x, y, z 데이터 포인트를 생성합니다.
• plt.axis('off') 메서드로 축 속성을 설정하여 기본 축을 숨깁니다.
• x, y, z 데이터를 인자로 contourf() 메서드를 호출해 등고선 플롯을 그립니다.
• axhline(0)과 axvline(0)으로 원점을 지나는 x=0, y=0 선을 빨간색으로 표시합니다.
• show() 메서드로 그림을 화면에 출력합니다.

예제 코드

import numpy as np
import matplotlib.pyplot as plt

plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True

x = np.linspace(-1.0, 1.0, 10)
x, y = np.meshgrid(x, x)
z = -np.hypot(x, y)

plt.axis('off')
plt.contourf(x, y, z, 10)
plt.axhline(0, color='red')
plt.axvline(0, color='red')

plt.show()

실행 결과

위 코드를 실행하면 등고선 플롯 위에 원점(0, 0)을 통과하는 빨간색 수평선과 수직선이 함께 표시됩니다. 이 방식은 데이터의 중심축이나 기준선을 시각적으로 강조하고 싶을 때 유용하게 활용할 수 있습니다.