156 KiB
156 KiB
None
<html lang="en">
<head>
gitea_input694149577
</head>
</html>
In [2]:
import polars as ps
pokémon = ps.read_csv("./pokémon/Pokemon.csv")
pokémon
Out[2]:
In [3]:
pokémon.sort(ps.col("HP")).tail(1)
Out[3]:
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]:
In [5]:
gen1 = pnn.filter(col("Generation") == 1)
gen1
Out[5]:
In [6]:
no_megas = pnn.filter(~col("Name").str.contains("Mega "))
no_megas
Out[6]:
In [7]:
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")
)
In [8]:
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")
)
In [22]:
types = pokémon.select(["Type 1"]).unique().to_series().to_list()
In [25]:
pokétype = ps.read_csv("./pokémon/Pokemon Type Chart.csv")
pokétype
Out[25]:
In [255]:
from dataclasses import dataclass
from enum import Enum
from typing import TypeVar
import random
TPokémon = TypeVar("TPokémon", bound="Pokémon")
pokémon_list: list = []
class Effective(Enum):
"""
Enum Description:
"""
SUPER = "It's super effective!"
NORMAL = ""
NOTVERY = "It's not very effective..."
NOT = "But it has no effect!"
@dataclass
class Pokémon:
name: str
types: list
hp: int
curr_hp: int
attack: int
defense: int
sp_atk: int
sp_def: int
def attack_opponent(self, opponent: TPokémon, typeatk: str, atkpwr: int):
if opponent.curr_hp == 0:
print("You cannot attack a fainted Pokémon!")
return
type_modifier = pokétype.filter(pokétype[""].str.contains(typeatk))[opponent.types[0]][0]
if (opponent.types[1] is not None):
type_modifier *= pokétype.filter(pokétype[""].str.contains(typeatk))[opponent.types[1]][0]
damage = self.attack * type_modifier * (atkpwr / 100)
if(typeatk in self.types):
damage = damage * 1.5
opponent.curr_hp = opponent.curr_hp - damage
print(f"""{self.name} attacks {opponent.name} for {int(damage)} damage!\n
{Effective.SUPER.value if type_modifier > 1.0 else Effective.NORMAL.value}
{Effective.NOTVERY.value if ((type_modifier < 1.0) and (type_modifier > 0.0)) else Effective.NORMAL.value}
{Effective.NOT.value if type_modifier == 0.0 else Effective.NORMAL.value}""")
if (opponent.curr_hp <= 0):
opponent.curr_hp = 0
print(f"{opponent.name} fainted!")
return
print(f"It now has {int(opponent.curr_hp)} HP")
def heal(self):
self.curr_hp = self.hp
for row in pokémon.rows(named=True):
pokémon_list.append(Pokémon(
name = row["Name"],
types = [row["Type 1"], row["Type 2"]],
hp = row["HP"],
curr_hp = row["HP"],
attack = row["Attack"],
defense = row["Defense"],
sp_atk = row["Sp. Atk"],
sp_def = row["Sp. Def"],
))
In [256]:
pokémon_list[0].attack_opponent(pokémon_list[15], "Grass", 40)
In [201]:
for poke in pokémon_list:
poke.heal()