Allow parametrizing paths

This commit is contained in:
Aarni Koskela
2022-08-31 15:12:46 +03:00
parent cdc6d9cc89
commit 538bc6083a
6 changed files with 19 additions and 8 deletions

View File

@@ -3,6 +3,8 @@ import re
import numpy as np
import pandas as pd
from pulkka.config import DATA_DIR
COLUMN_MAP = {
"Missä kaupungissa työpaikkasi pääasiallinen toimisto sijaitsee?": "Kaupunki",
"Työaika (jos työsuhteessa)": "Työaika",
@@ -61,7 +63,7 @@ def map_ika(d):
def read_data() -> pd.DataFrame:
df: pd.DataFrame = pd.read_excel(
"data/results.xlsx",
DATA_DIR / "results.xlsx",
skiprows=[1], # Google Sheets exports one empty row
)
df.rename(columns=COLUMN_MAP, inplace=True)