Jupyter-ITVitae/pokémon.ipynb at 2e9f7975d13f200f13bd3ef9b54d85e18d129a6f

In [2]:

import polars as ps

pokémon = ps.read_csv("./pokémon/Pokemon.csv")
pokémon

Out[2]:

shape: (800, 13)

#	Name	Type 1	Type 2	Total	HP	Attack	Defense	Sp. Atk	Sp. Def	Speed	Generation	Legendary
i64	str	str	str	i64	i64	i64	i64	i64	i64	i64	i64	bool
1	"Bulbasaur"	"Grass"	"Poison"	318	45	49	49	65	65	45	1	false
2	"Ivysaur"	"Grass"	"Poison"	405	60	62	63	80	80	60	1	false
3	"Venusaur"	"Grass"	"Poison"	525	80	82	83	100	100	80	1	false
3	"VenusaurMega V…	"Grass"	"Poison"	625	80	100	123	122	120	80	1	false
4	"Charmander"	"Fire"	null	309	39	52	43	60	50	65	1	false
5	"Charmeleon"	"Fire"	null	405	58	64	58	80	65	80	1	false
6	"Charizard"	"Fire"	"Flying"	534	78	84	78	109	85	100	1	false
6	"CharizardMega …	"Fire"	"Dragon"	634	78	130	111	130	85	100	1	false
6	"CharizardMega …	"Fire"	"Flying"	634	78	104	78	159	115	100	1	false
7	"Squirtle"	"Water"	null	314	44	48	65	50	64	43	1	false
8	"Wartortle"	"Water"	null	405	59	63	80	65	80	58	1	false
9	"Blastoise"	"Water"	null	530	79	83	100	85	105	78	1	false
…	…	…	…	…	…	…	…	…	…	…	…	…
712	"Bergmite"	"Ice"	null	304	55	69	85	32	35	28	6	false
713	"Avalugg"	"Ice"	null	514	95	117	184	44	46	28	6	false
714	"Noibat"	"Flying"	"Dragon"	245	40	30	35	45	40	55	6	false
715	"Noivern"	"Flying"	"Dragon"	535	85	70	80	97	80	123	6	false
716	"Xerneas"	"Fairy"	null	680	126	131	95	131	98	99	6	true
717	"Yveltal"	"Dark"	"Flying"	680	126	131	95	131	98	99	6	true
718	"Zygarde50% For…	"Dragon"	"Ground"	600	108	100	121	81	95	95	6	true
719	"Diancie"	"Rock"	"Fairy"	600	50	100	150	100	150	50	6	true
719	"DiancieMega Di…	"Rock"	"Fairy"	700	50	160	110	160	110	110	6	true
720	"HoopaHoopa Con…	"Psychic"	"Ghost"	600	80	110	60	150	130	70	6	true
720	"HoopaHoopa Unb…	"Psychic"	"Dark"	680	80	160	60	170	130	80	6	true
721	"Volcanion"	"Fire"	"Water"	600	80	110	120	130	90	70	6	true

In [3]:

pokémon.sort(ps.col("HP")).tail(1)

Out[3]:

shape: (1, 13)

#	Name	Type 1	Type 2	Total	HP	Attack	Defense	Sp. Atk	Sp. Def	Speed	Generation	Legendary
i64	str	str	str	i64	i64	i64	i64	i64	i64	i64	i64	bool
242	"Blissey"	"Normal"	null	540	255	10	10	75	135	55	2	false

In [4]:

from polars import col

pokémon.select(ps.col("Type 2").fill_null(ps.col("Type 1")))
pnn = pokémon.with_columns([
    ps.when(col("Type 2").is_null()).then(col("Type 1")).otherwise(col("Type 2")).alias("Type 2")
])

pnn

Out[4]:

shape: (800, 13)

#	Name	Type 1	Type 2	Total	HP	Attack	Defense	Sp. Atk	Sp. Def	Speed	Generation	Legendary
i64	str	str	str	i64	i64	i64	i64	i64	i64	i64	i64	bool
1	"Bulbasaur"	"Grass"	"Poison"	318	45	49	49	65	65	45	1	false
2	"Ivysaur"	"Grass"	"Poison"	405	60	62	63	80	80	60	1	false
3	"Venusaur"	"Grass"	"Poison"	525	80	82	83	100	100	80	1	false
3	"VenusaurMega V…	"Grass"	"Poison"	625	80	100	123	122	120	80	1	false
4	"Charmander"	"Fire"	"Fire"	309	39	52	43	60	50	65	1	false
5	"Charmeleon"	"Fire"	"Fire"	405	58	64	58	80	65	80	1	false
6	"Charizard"	"Fire"	"Flying"	534	78	84	78	109	85	100	1	false
6	"CharizardMega …	"Fire"	"Dragon"	634	78	130	111	130	85	100	1	false
6	"CharizardMega …	"Fire"	"Flying"	634	78	104	78	159	115	100	1	false
7	"Squirtle"	"Water"	"Water"	314	44	48	65	50	64	43	1	false
8	"Wartortle"	"Water"	"Water"	405	59	63	80	65	80	58	1	false
9	"Blastoise"	"Water"	"Water"	530	79	83	100	85	105	78	1	false
…	…	…	…	…	…	…	…	…	…	…	…	…
712	"Bergmite"	"Ice"	"Ice"	304	55	69	85	32	35	28	6	false
713	"Avalugg"	"Ice"	"Ice"	514	95	117	184	44	46	28	6	false
714	"Noibat"	"Flying"	"Dragon"	245	40	30	35	45	40	55	6	false
715	"Noivern"	"Flying"	"Dragon"	535	85	70	80	97	80	123	6	false
716	"Xerneas"	"Fairy"	"Fairy"	680	126	131	95	131	98	99	6	true
717	"Yveltal"	"Dark"	"Flying"	680	126	131	95	131	98	99	6	true
718	"Zygarde50% For…	"Dragon"	"Ground"	600	108	100	121	81	95	95	6	true
719	"Diancie"	"Rock"	"Fairy"	600	50	100	150	100	150	50	6	true
719	"DiancieMega Di…	"Rock"	"Fairy"	700	50	160	110	160	110	110	6	true
720	"HoopaHoopa Con…	"Psychic"	"Ghost"	600	80	110	60	150	130	70	6	true
720	"HoopaHoopa Unb…	"Psychic"	"Dark"	680	80	160	60	170	130	80	6	true
721	"Volcanion"	"Fire"	"Water"	600	80	110	120	130	90	70	6	true

In [5]:

gen1 = pnn.filter(col("Generation") == 1)
gen1

Out[5]:

shape: (166, 13)

#	Name	Type 1	Type 2	Total	HP	Attack	Defense	Sp. Atk	Sp. Def	Speed	Generation	Legendary
i64	str	str	str	i64	i64	i64	i64	i64	i64	i64	i64	bool
1	"Bulbasaur"	"Grass"	"Poison"	318	45	49	49	65	65	45	1	false
2	"Ivysaur"	"Grass"	"Poison"	405	60	62	63	80	80	60	1	false
3	"Venusaur"	"Grass"	"Poison"	525	80	82	83	100	100	80	1	false
3	"VenusaurMega V…	"Grass"	"Poison"	625	80	100	123	122	120	80	1	false
4	"Charmander"	"Fire"	"Fire"	309	39	52	43	60	50	65	1	false
5	"Charmeleon"	"Fire"	"Fire"	405	58	64	58	80	65	80	1	false
6	"Charizard"	"Fire"	"Flying"	534	78	84	78	109	85	100	1	false
6	"CharizardMega …	"Fire"	"Dragon"	634	78	130	111	130	85	100	1	false
6	"CharizardMega …	"Fire"	"Flying"	634	78	104	78	159	115	100	1	false
7	"Squirtle"	"Water"	"Water"	314	44	48	65	50	64	43	1	false
8	"Wartortle"	"Water"	"Water"	405	59	63	80	65	80	58	1	false
9	"Blastoise"	"Water"	"Water"	530	79	83	100	85	105	78	1	false
…	…	…	…	…	…	…	…	…	…	…	…	…
142	"AerodactylMega…	"Rock"	"Flying"	615	80	135	85	70	95	150	1	false
143	"Snorlax"	"Normal"	"Normal"	540	160	110	65	65	110	30	1	false
144	"Articuno"	"Ice"	"Flying"	580	90	85	100	95	125	85	1	true
145	"Zapdos"	"Electric"	"Flying"	580	90	90	85	125	90	100	1	true
146	"Moltres"	"Fire"	"Flying"	580	90	100	90	125	85	90	1	true
147	"Dratini"	"Dragon"	"Dragon"	300	41	64	45	50	50	50	1	false
148	"Dragonair"	"Dragon"	"Dragon"	420	61	84	65	70	70	70	1	false
149	"Dragonite"	"Dragon"	"Flying"	600	91	134	95	100	100	80	1	false
150	"Mewtwo"	"Psychic"	"Psychic"	680	106	110	90	154	90	130	1	true
150	"MewtwoMega Mew…	"Psychic"	"Fighting"	780	106	190	100	154	100	130	1	true
150	"MewtwoMega Mew…	"Psychic"	"Psychic"	780	106	150	70	194	120	140	1	true
151	"Mew"	"Psychic"	"Psychic"	600	100	100	100	100	100	100	1	false

In [6]:

no_megas = pnn.filter(~col("Name").str.contains("Mega "))
no_megas

Out[6]:

shape: (752, 13)

#	Name	Type 1	Type 2	Total	HP	Attack	Defense	Sp. Atk	Sp. Def	Speed	Generation	Legendary
i64	str	str	str	i64	i64	i64	i64	i64	i64	i64	i64	bool
1	"Bulbasaur"	"Grass"	"Poison"	318	45	49	49	65	65	45	1	false
2	"Ivysaur"	"Grass"	"Poison"	405	60	62	63	80	80	60	1	false
3	"Venusaur"	"Grass"	"Poison"	525	80	82	83	100	100	80	1	false
4	"Charmander"	"Fire"	"Fire"	309	39	52	43	60	50	65	1	false
5	"Charmeleon"	"Fire"	"Fire"	405	58	64	58	80	65	80	1	false
6	"Charizard"	"Fire"	"Flying"	534	78	84	78	109	85	100	1	false
7	"Squirtle"	"Water"	"Water"	314	44	48	65	50	64	43	1	false
8	"Wartortle"	"Water"	"Water"	405	59	63	80	65	80	58	1	false
9	"Blastoise"	"Water"	"Water"	530	79	83	100	85	105	78	1	false
10	"Caterpie"	"Bug"	"Bug"	195	45	30	35	20	20	45	1	false
11	"Metapod"	"Bug"	"Bug"	205	50	20	55	25	25	30	1	false
12	"Butterfree"	"Bug"	"Flying"	395	60	45	50	90	80	70	1	false
…	…	…	…	…	…	…	…	…	…	…	…	…
711	"GourgeistSuper…	"Ghost"	"Grass"	494	85	100	122	58	75	54	6	false
712	"Bergmite"	"Ice"	"Ice"	304	55	69	85	32	35	28	6	false
713	"Avalugg"	"Ice"	"Ice"	514	95	117	184	44	46	28	6	false
714	"Noibat"	"Flying"	"Dragon"	245	40	30	35	45	40	55	6	false
715	"Noivern"	"Flying"	"Dragon"	535	85	70	80	97	80	123	6	false
716	"Xerneas"	"Fairy"	"Fairy"	680	126	131	95	131	98	99	6	true
717	"Yveltal"	"Dark"	"Flying"	680	126	131	95	131	98	99	6	true
718	"Zygarde50% For…	"Dragon"	"Ground"	600	108	100	121	81	95	95	6	true
719	"Diancie"	"Rock"	"Fairy"	600	50	100	150	100	150	50	6	true
720	"HoopaHoopa Con…	"Psychic"	"Ghost"	600	80	110	60	150	130	70	6	true
720	"HoopaHoopa Unb…	"Psychic"	"Dark"	680	80	160	60	170	130	80	6	true
721	"Volcanion"	"Fire"	"Water"	600	80	110	120	130	90	70	6	true

In [24]:

from plotnine import ggplot, aes, geom_histogram, labs

(
    ggplot(no_megas.to_pandas())
    + aes(x="Attack") 
    + geom_histogram(bins=(200/25))
    + labs(title="Distribution of attack stat for all Pokémon", y="Amount of Pokémon")
)

No description has been provided for this image

In [28]:

only_fairy = pnn.filter(col("Type 1").str.contains("Fairy") | col("Type 2").str.contains("Fairy"))

(
    ggplot(only_fairy.to_pandas())
    + aes(x="Generation") 
    + geom_histogram(bins=(6), color='#EF9FEF', fill='#EF6FEF')
    + labs(title="Amount of fairy Pokémon in each generation", y="Amount of Pokémon")
)

145 KiB Raw Blame History

145 KiB

Raw Blame History