파이썬의 Matplotlib 라이브러리를 사용하면 산점도(scatter plot) 위에 원하는 선을 손쉽게 추가할 수 있습니다. 이 글에서는 산점도에 선을 그리는 방법을 단계별로 살펴보겠습니다.
산점도에 선 추가하는 단계
- 그림(figure) 크기를 설정하고 서브플롯 주변 및 사이의 여백(padding)을 조정합니다.
- 데이터 포인트 개수를 저장할 변수 n을 초기화합니다.
- scatter() 메서드를 사용해 x, y 데이터 포인트를 산점도로 표현합니다.
- plot() 메서드를 사용하여 선을 그립니다.
- xlim() 메서드로 X축의 범위를 제한합니다.
- show() 메서드를 호출하여 그림을 화면에 표시합니다.
예제 코드
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 = np.random.rand(n) y = np.random.rand(n) plt.scatter(x, y, c=x) plt.plot([0.1, 0.4, 0.3, 0.2]) plt.xlim(0, 1) plt.show()
실행 결과
위 코드를 실행하면 무작위로 생성된 100개의 데이터 포인트가 색상별로 표시된 산점도와 함께, 지정한 값들을 연결하는 선이 같은 좌표 평면 위에 함께 나타납니다. X축은 0부터 1까지의 범위로 제한되어 출력됩니다.
이처럼 scatter()와 plot() 메서드를 함께 사용하면 데이터 분포와 추세선 또는 기준선을 하나의 그래프에서 동시에 시각화할 수 있어, 데이터 분석과 시각화 작업에 매우 유용합니다.