Computer >> 컴퓨터 >  >> 프로그래밍 >> Python

Python Matplotlib plt.title()에 변수를 추가하는 방법

Python의 plt.title() 함수에 변수를 포함하려면 문자열 포맷팅 기법을 활용하면 됩니다. 아래 단계를 따라 진행할 수 있습니다.

  • numpy를 사용해 x와 y 데이터 포인트를 생성하고, 변수 num으로 y 값을 계산한 뒤 이를 그래프 제목에 반영합니다.
  • plot() 메서드를 사용해 x와 y 데이터 포인트를 빨간색으로 그립니다.
  • 변수 num을 활용해 곡선의 제목을 동적으로 설정합니다.
  • show() 메서드를 호출하여 완성된 그래프를 화면에 표시합니다.

예제 코드

import numpy as np
from matplotlib import pyplot as plt
plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True
x = np.linspace(-1, 1, 10)
num = 2
y = num ** x
plt.plot(x, y, c='red')
plt.title(f"y=%d$^x$" % num)
plt.show()

출력 결과

위 코드를 실행하면 빨간색 지수 곡선이 그려지며, 그래프 상단에는 변수 num 값이 적용된 "y=2x"라는 제목이 표시됩니다.

다른 방법: f-string 활용

% 포맷팅 대신 f-string을 사용하면 코드가 더 간결하고 가독성이 좋아집니다. 위 예제에서 제목 설정 부분을 다음과 같이 바꿀 수 있습니다.

plt.title(f"y={num}$^x$")

f-string은 Python 3.6 이상에서 지원되며, 중괄호 {} 안에 변수 이름을 직접 작성하므로 여러 변수를 조합하거나 복잡한 제목을 만들 때 특히 유용합니다. 또한 LaTeX 문법($^x$ 등)을 함께 사용하면 위첨자 같은 수학 기호도 제목에 자연스럽게 표현할 수 있습니다.