From 61093bde65514dc161c997be472841376d0de220 Mon Sep 17 00:00:00 2001 From: Aarni Koskela Date: Fri, 19 Feb 2021 15:02:06 +0200 Subject: [PATCH] =?UTF-8?q?Fix=20mis-bracket=20in=20Ik=C3=A4?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- data_ingest.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/data_ingest.py b/data_ingest.py index ecb41d1..9e69720 100644 --- a/data_ingest.py +++ b/data_ingest.py @@ -53,6 +53,12 @@ def map_numberlike(d): return d +def map_ika(d): + if d == "30-35 v": # Early answers had a wrong bracket here + d = "31-35 v" + return d + + def read_data() -> pd.DataFrame: df: pd.DataFrame = pd.read_excel( "data/results.xlsx", @@ -65,7 +71,7 @@ def read_data() -> pd.DataFrame: ) df["Kaupunki"] = df["Kaupunki"].astype("category") df["Sukupuoli"] = df["Sukupuoli"].apply(map_sukupuoli).astype("category") - df["Ikä"] = df["Ikä"].astype("category") + df["Ikä"] = df["Ikä"].apply(map_ika).astype("category") df["Etä"] = df["Etä"].map(ETATYO_MAP).astype("category") df["Kilpailukykyinen"].replace({"Kyllä": True, "Ei": False}, inplace=True)