Matplotlib) 데이터 시각화 7편, 산점도 그래프

나도코딩 유튜브를 참고하여 코드를 만들었습니다. 

 

---

import matplotlib.pyplot as plt
import matplotlib
import pandas as pd
import numpy as np


matplotlib.rcParams['font.family'] = 'Malgun Gothic'
matplotlib.rcParams['font.size'] = 15 
matplotlib.rcParams['axes.unicode_minus'] = False

# 산점도 그래프
df = pd.read_excel('score.xlsx')
df['학년'] = [3,3,2,1,1,3,2,2]

sizes = df['학년'] *500 #1학년 500 2학년 1000 3학년 1500
#c로 학년 나누고 cmap으로 학년별 색깔 골라주기! 
plt.figure(figsize=(10,10))
plt.scatter(df['영어'],df['수학'],s = sizes, c = df['학년'], cmap ='viridis', alpha = 0.3)

plt.xlabel('영어 점수')
plt.ylabel('수학 점수')
plt.colorbar(ticks = [1,2,3],label = '학년',shrink= 0.8, orientation = 'horizontal')
plt.show()

c, cmap을 통해서 산점도 그래프를 더 이쁘게 만들 수 있습니다.