Matplotlib은 파이썬에서 데이터 시각화를 위한 강력한 라이브러리입니다. 산점도(Scatter Plot)는 데이터간의 상관 관계를 시각적으로 표현하는 방법 중 하나입니다. 이번 글에서는 Matplotlib를 사용하여 파이썬에서 산점도를 그리는 방법에 대해 알아보겠습니다.

1. Matplotlib 설치

Matplotlib는 대부분의 파이썬 환경에서 기본적으로 설치되어 있지만, 설치되어 있지 않다면 아래 명령어로 설치할 수 있습니다.

pip install matplotlib

2. 데이터 준비

먼저, 예제를 위해 임의의 데이터를 만들어보겠습니다. x축과 y축에 각각 100개의 무작위 숫자를 생성하여 리스트로 저장합니다.

import random

x = [random.randint(0, 100) for _ in range(100)]
y = [random.randint(0, 100) for _ in range(100)]

3. 산점도 그리기

Matplotlib의 scatter 함수를 사용하여 산점도를 그릴 수 있습니다. scatter 함수는 x축 데이터와 y축 데이터를 인자로 받습니다.

import matplotlib.pyplot as plt

plt.scatter(x, y)
plt.show()

위 코드를 실행하면, 새로운 창이 뜨면서 x축과 y축에 해당하는 데이터를 가지고 산점도가 그려집니다.

4. 그래프 꾸미기

위의 예제는 기본적인 산점도를 그리는 방법이며, 추가적으로 그래프를 꾸밀 수 있는 다양한 옵션이 제공됩니다.

• color: 산점도 점의 색상을 설정할 수 있습니다.
• s: 산점도 점의 크기를 설정할 수 있습니다.
• alpha: 산점도 점의 투명도를 설정할 수 있습니다.

plt.scatter(x, y, color='red', s=50, alpha=0.5)
plt.title('Scatter Plot')
plt.xlabel('X')
plt.ylabel('Y')
plt.grid(True)
plt.show()

위 코드는 점의 색상을 빨간색으로, 크기를 50으로, 투명도를 0.5로 설정한 산점도를 그리고 제목과 x축, y축 레이블을 추가하고, 그리드를 표시합니다.

Matplotlib를 사용하여 파이썬에서 산점도를 그리는 방법을 알아봤습니다. 산점도는 데이터의 분포와 상관 관계를 빠르게 파악할 수 있는 강력한 시각화 도구입니다. Matplotlib의 다양한 옵션을 활용하여 데이터를 더욱 잘 표현할 수 있습니다.