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

0913

4주차-Day3) EDA

EDA 실습 미션

'포켓몬' 데이터셋에서의 다양한 기준을 활용한 그래프 만들어보기
# 미션 수행한 것 정리

https://www.kaggle.com/datasets/abcsds/pokemon
사용한 데이터셋 링크

# 라이브러리 불러오기

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

%matplotlib inline

pokemon_df = pd.read_csv("./Pokemon.csv")

#  포켓몬 타입에 따른 방어력 히트맵으로 나타내기
sns.heatmap(pokemon_df[['Type 1', 'Defense']].groupby(['Type 1']).mean())
plt.plot()

#  포켓몬 타입에 따른 스피드 그래프로 나타내기
for i in pokemon_df['Type 1'].unique():
    pokemon_df['Speed'][pokemon_df['Type 1'] == i].plot(kind='kde')
plt.show()

#  포켓몬 타입에 따른 공격력 그래프로 나타내기
pokemon_type = pokemon_df['Attack'].groupby(by=pokemon_df['Type 1'])
pokemon_type

fig = plt.figure(figsize=(10,5))
plt.plot(pokemon_type.mean())
plt.xticks(rotation=90)
plt.title('Attck')
plt.show()