맷플롯립(matplotlib) 패키지로 산점도(scatter plot) 그리기

파이썬에서 과학계산용 그래프를 그리는 대표적인 패키지는 맷플롯립이다.

 

이 패키지를 임포트하고 산점도를 그리는 scatter() 함수를 사용해본다.

 

도미와 빙어를 무게, 길이 데이터로 산점도를 그린다.

 

# 도미의 길이, 무게 데이터
bream_length = [25.4, 26.3, 26.5, 29.0, 29.0, 29.7, 29.7, 30.0, 30.0, 30.7, 31.0, 31.0, 
                31.5, 32.0, 32.0, 32.0, 33.0, 33.0, 33.5, 33.5, 34.0, 34.0, 34.5, 35.0, 
                35.0, 35.0, 35.0, 36.0, 36.0, 37.0, 38.5, 38.5, 39.5, 41.0, 41.0]
bream_weight = [242.0, 290.0, 340.0, 363.0, 430.0, 450.0, 500.0, 390.0, 450.0, 500.0, 475.0, 500.0, 
                500.0, 340.0, 600.0, 600.0, 700.0, 700.0, 610.0, 650.0, 575.0, 685.0, 620.0, 680.0, 
                700.0, 725.0, 720.0, 714.0, 850.0, 1000.0, 920.0, 955.0, 925.0, 975.0, 950.0]



#맷플롯립의 pyplot 함수를 plt로 임포트
import matplotlib.pyplot as plt



plt.scatter(bream_length, bream_weight) #산점도 그래프 생성
plt.xlabel('length') #x축은 길이
plt.ylabel('weight') #y축은 무게
plt.show() #그래프 출력 1


# 빙어의 길이, 무게 데이터
smelt_length = [9.8, 10.5, 10.6, 11.0, 11.2, 11.3, 11.8, 11.8, 12.0, 12.2, 12.4, 13.0, 14.3, 15.0]
smelt_weight = [6.7, 7.5, 7.0, 9.7, 9.8, 8.7, 10.0, 9.9, 9.8, 12.2, 13.4, 12.2, 19.7, 19.9]



plt.scatter(bream_length, bream_weight) #산점도 그래프 생성1
plt.scatter(smelt_length, smelt_weight) #산점도 그래프 생성2. 생성시 1, 2가 같이 나온다
plt.xlabel('length') #x축은 길이
plt.ylabel('weight') #y축은 무게
plt.show() #그래프 출력 2

 

결과는 다음과 같다

 

그래프 출력 1

 

그래프 출력 2