[6기] 프로그래머스 인공지능 데브코스 16일차 TIL

0916

4주차-Day5) Mission 2. Abengers, Assemble!

EDA 실습 미션

사용된 데이터셋
kaggle.com/dannielr/marvel-superheroes?select=charcters_stats.csv

# 라이브러리 불러오기

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns
import json

%matplotlib inline

mv = pd.read_csv("./charcters_stats.csv")

# 수치의 합이 가장 큰 캐릭터 정보 
mv.loc[mv["Total"].idxmax()]


# 좋은 캐릭터와 나쁜 캐릭터의 능력치들의 수치 분포
mv_good = []
mv_bad = []

fig = plt.figure(figsize=(10,5))

for i in mv.columns[2:8]:
    mv_good.append(mv[i].groupby(mv['Alignment']).mean()[0])
    mv_bad.append(mv[i].groupby(mv['Alignment']).mean()[1])
plt.plot(mv_good, color='#f13900', label='good_heros')  # 선 그래프로 평균들 표현
plt.plot(mv_bad, color='#003280', label='bad_heros')
ticklabel=list(mv.columns[2:8])
plt.xticks(np.arange(0, 6, 1), ticklabel)
plt.legend()
plt.title('Status')
plt.show()