import pandas as pd

# === Cesty k souborům ===
vodafone_path = r"C:\Škola\BAK\AnalytickaCast\Kamdojizdime\PripavaDat\9.PrumerNoc\9.VodafoneDataLongFormat.xlsx"
bp_path = r"C:\Škola\BAK\AnalytickaCast\Kamdojizdime\PripavaDat\9.PrumerNoc\9.KamDojizdimeKomplet.xlsx"
output_path = r"C:\Škola\BAK\AnalytickaCast\Kamdojizdime\PripavaDat\9.PrumerNoc\Vystup_Prumery_Rezidenti_NOC.xlsx"

# === Noční časové intervaly (22:00–06:00) ===
nocni_intervaly = [
    "22:00 - 23:00", "23:00 - 24:00", "00:00 - 01:00",
    "01:00 - 02:00", "02:00 - 03:00", "03:00 - 04:00",
    "04:00 - 05:00", "05:00 - 06:00"
]

# === Vodafone ===
df_vodafone = pd.read_excel(vodafone_path)
df_vodafone_noc = df_vodafone[
    df_vodafone["Časový interval"].isin(nocni_intervaly)
]
df_vodafone_agg = df_vodafone_noc.groupby("Kod_obec")["Rezidenti"].mean().reset_index()
df_vodafone_agg = df_vodafone_agg.rename(columns={"Rezidenti": "Rezidenti_Vodafone_noc"})

# === KamDojizdime ===
df_bp = pd.read_excel(bp_path)
df_bp_noc = df_bp[df_bp["Časový interval"].isin(nocni_intervaly)]
df_bp_agg = df_bp_noc.groupby("Kod_obec")["Rezidenti"].mean().reset_index()
df_bp_agg = df_bp_agg.rename(columns={"Rezidenti": "Rezidenti_BP_noc"})

# === Sjednocení dle Kod_obec ===
df_merged = pd.merge(df_vodafone_agg, df_bp_agg, on="Kod_obec", how="outer")

# === Uložení výstupu ===
df_merged.to_excel(output_path, index=False)
print(" Hotovo – výstup uložen jako:", output_path)

# skript byl optimalizován pomocí nástroje ChatGPT 4.0
