265 lines
8.3 KiB
Python
Executable File
265 lines
8.3 KiB
Python
Executable File
import pandas as pd
|
||
import matplotlib.pyplot as plt
|
||
import os
|
||
import socket
|
||
|
||
def interpolate_daily(df):
|
||
df = df.copy()
|
||
df["Datum"] = pd.to_datetime(df["Datum"], errors="coerce")
|
||
df = df.sort_values("Datum").set_index("Datum")
|
||
|
||
daily_index = pd.date_range(df.index.min(), df.index.max(), freq="D")
|
||
|
||
return (
|
||
df.reindex(daily_index)
|
||
.interpolate(method="time")
|
||
.reset_index()
|
||
.rename(columns={"index": "Datum"})
|
||
)
|
||
|
||
def plot_percentiles(ax, x, data_dict, p_keys, colors=None, linewidths=None, labels=None,
|
||
linestyle=None, fill=True, fill_color="lightcoral", fill_alpha=0.25):
|
||
"""
|
||
Plot multiple percentile curves from a dictionary like who_head_daily.
|
||
|
||
Parameters
|
||
----------
|
||
ax : matplotlib axis
|
||
x : array-like (x values)
|
||
data_dict : dict-like, e.g. who_head_daily
|
||
p_keys : list of keys, e.g. ["P3", "P50", "P97"]
|
||
colors : list of colors (optional)
|
||
linewidths : list of linewidths (optional)
|
||
fill : whether to fill between min and max percentile
|
||
:param labels:
|
||
"""
|
||
|
||
n = len(x)
|
||
|
||
if colors is None:
|
||
colors = [None] * len(p_keys)
|
||
if linewidths is None:
|
||
linewidths = [1.5] * len(p_keys)
|
||
|
||
# Plot lines
|
||
for i, key in enumerate(p_keys):
|
||
ax.plot(
|
||
x,
|
||
data_dict[key][:n],
|
||
color=colors[i],
|
||
linewidth=linewidths[i],
|
||
label=labels[i],
|
||
linestyle=linestyle[i]
|
||
)
|
||
|
||
# Fill between min and max percentile
|
||
if fill and len(p_keys) >= 2:
|
||
lower = data_dict[p_keys[0]][:n]
|
||
upper = data_dict[p_keys[-1]][:n]
|
||
|
||
ax.fill_between(
|
||
x,
|
||
lower,
|
||
upper,
|
||
color=fill_color,
|
||
alpha=fill_alpha
|
||
)
|
||
|
||
return ax
|
||
|
||
hostname = socket.gethostname()
|
||
if hostname == "quiethost":
|
||
|
||
# Build output path from ods_path
|
||
ods_path = "/data/Nextcloud/Documents/Dokumente/Shared/Alina/Wachstum/WachstumsDatenWHO.ods"
|
||
csv_path = "/data/Nextcloud/Documents/Dokumente/Shared/Alina/Wachstum/WachstumAlina.csv"
|
||
elif hostname == "serverhost":
|
||
print("Serverhost Mode")
|
||
import subprocess
|
||
|
||
subprocess.run(["bash", "/home/lucas/scripts/serverhost/syncNC.sh", "-al"])
|
||
ods_path = "/home/lucas/Documents/Dokumente/Shared/Alina/Wachstum/WachstumsDatenWHO.ods"
|
||
csv_path = "/home/lucas/Documents/Dokumente/Shared/Alina/Wachstum/WachstumAlina.csv"
|
||
output_dir = os.path.dirname(ods_path)
|
||
output_path = os.path.join(output_dir, "WachstumAlina.png")
|
||
|
||
# Load specific sheets into data structures (DataFrames)
|
||
alina_df = pd.read_csv(csv_path)
|
||
|
||
who_weight_df = pd.read_excel(ods_path,sheet_name="WHO Gewicht",engine="odf")
|
||
who_height_df = pd.read_excel(ods_path,sheet_name="WHO Größe",engine="odf")
|
||
who_head_df = pd.read_excel(ods_path,sheet_name="WHO Kopfumfang",engine="odf")
|
||
|
||
def calculateDateFromBrithday(df, birthday ):
|
||
|
||
new_df = pd.DataFrame(columns=["Datum"])
|
||
for idx, (val1, val2) in enumerate(zip(df["Month"], df["Week"])):
|
||
if not pd.isna(val1):
|
||
new_df.loc[idx, "Datum"] = (birthday + pd.DateOffset(months=val1))
|
||
if not pd.isna(val2):
|
||
new_df.loc[idx, "Datum"] = (birthday + pd.DateOffset(weeks=val2))
|
||
|
||
return pd.to_datetime(new_df["Datum"], format="%d.%m.%Y", errors="coerce")
|
||
|
||
alina_df["Datum"] = pd.to_datetime(alina_df["Datum"],format="%d.%m.%Y",errors="coerce")
|
||
|
||
# Ensure Datum is datetime (safe to run even if already converted)
|
||
who_weight_df["Datum"] = calculateDateFromBrithday(who_weight_df,alina_df["Datum"][0] )
|
||
who_height_df["Datum"] = calculateDateFromBrithday(who_height_df,alina_df["Datum"][0] )
|
||
who_head_df["Datum"] = calculateDateFromBrithday(who_head_df,alina_df["Datum"][0] )
|
||
|
||
who_weight_daily = interpolate_daily(who_weight_df)
|
||
who_height_daily = interpolate_daily(who_height_df)
|
||
who_head_daily = interpolate_daily(who_head_df)
|
||
|
||
print(alina_df["Datum"])
|
||
# Plot
|
||
last_alina_date = alina_df["Datum"].max() + pd.DateOffset(months=2)
|
||
x = who_weight_daily["Datum"][who_weight_daily["Datum"] <= last_alina_date]
|
||
fig, axes = plt.subplots(
|
||
nrows=3,
|
||
ncols=1,
|
||
figsize=(12, 10),
|
||
sharex=True
|
||
)
|
||
|
||
percentile_list = ['P1', 'P3', 'P10', 'P25',
|
||
'P50', 'P75', 'P90', 'P97', 'P99']
|
||
colors_list = ["lightcoral" if p != "P50" else "darkred" for p in percentile_list]
|
||
linewidths_list = [1.5 if p != "P50" else 2.5 for p in percentile_list]
|
||
label_list = [p if p == "P50" or p == "P3" or p == "P97" else "" for p in percentile_list]
|
||
linestyle_list = ["--" if p == "P3" or p == "P97" else "-" for p in percentile_list]
|
||
# ---------- 1. WEIGHT ----------
|
||
ax = axes[0]
|
||
plot_percentiles(
|
||
ax,
|
||
x,
|
||
who_weight_daily,
|
||
percentile_list,
|
||
colors=colors_list,
|
||
linewidths=linewidths_list,
|
||
labels=label_list,
|
||
linestyle=linestyle_list
|
||
)
|
||
|
||
ax.plot(alina_df["Datum"], alina_df["Gewicht"].interpolate(limit_area="inside"),
|
||
color="green", linewidth=1.5, marker="o", label="Alina")
|
||
|
||
last_idx = alina_df["Gewicht"].last_valid_index()
|
||
who_row = who_weight_daily.loc[who_weight_daily["Datum"] == alina_df["Datum"][last_idx]].iloc[0]
|
||
perzentil = (alina_df["Gewicht"][last_idx]-who_row["P50"])/((who_row["P97"]-who_row["P50"])/47)+50
|
||
|
||
ax.text(alina_df["Datum"].iloc[-1]+pd.DateOffset(days=1), alina_df["Gewicht"].iloc[-1]+0.5,f"{perzentil:.2f} %",
|
||
fontsize=10,
|
||
ha="left",
|
||
va="bottom",
|
||
color="green")
|
||
|
||
ax.set_ylabel("Gewicht [kg]")
|
||
ax.set_title("Wachstum")
|
||
# Enable minor ticks
|
||
ax.minorticks_on()
|
||
|
||
# Configure major grid (e.g., solid, darker)
|
||
ax.grid(which='major', color='black', linestyle='-', linewidth=0.8)
|
||
|
||
# Configure minor grid (e.g., dashed, lighter, thinner)
|
||
ax.grid(which='minor', color='gray', linestyle='--', linewidth=0.4, alpha=0.5)
|
||
ax.legend()
|
||
|
||
|
||
# ---------- 2. HEIGHT ----------
|
||
ax = axes[1]
|
||
plot_percentiles(
|
||
ax,
|
||
x,
|
||
who_height_daily,
|
||
percentile_list,
|
||
colors=colors_list,
|
||
linewidths=linewidths_list,
|
||
labels=label_list,
|
||
linestyle=linestyle_list
|
||
)
|
||
ax.plot(alina_df["Datum"], alina_df["Größe"].interpolate(limit_area="inside"),color="green", linewidth=1.5, marker="o")
|
||
|
||
last_idx = alina_df["Größe"].last_valid_index()
|
||
who_row = who_height_daily.loc[who_height_daily["Datum"] == alina_df["Datum"][last_idx]].iloc[0]
|
||
perzentil = (alina_df["Größe"][last_idx]-who_row["P50"])/((who_row["P97"]-who_row["P50"])/47)+50
|
||
|
||
ax.text(alina_df["Datum"].iloc[-1]+pd.DateOffset(days=1), alina_df["Größe"].iloc[-1]+0.5,f"{perzentil:.2f} %",
|
||
fontsize=10,
|
||
ha="left",
|
||
va="bottom",
|
||
color="green")
|
||
|
||
|
||
ax.set_ylabel("Größe [cm]")
|
||
# Enable minor ticks
|
||
ax.minorticks_on()
|
||
|
||
# Configure major grid (e.g., solid, darker)
|
||
ax.grid(which='major', color='black', linestyle='-', linewidth=0.8)
|
||
# Configure minor grid (e.g., dashed, lighter, thinner)
|
||
ax.grid(which='minor', color='gray', linestyle='--', linewidth=0.4, alpha=0.5)
|
||
|
||
|
||
# ---------- 3. HEAD CIRCUMFERENCE ----------
|
||
ax = axes[2]
|
||
|
||
plot_percentiles(
|
||
ax,
|
||
x,
|
||
who_head_daily,
|
||
percentile_list,
|
||
colors=colors_list,
|
||
linewidths=linewidths_list,
|
||
labels=label_list,
|
||
linestyle=linestyle_list
|
||
)
|
||
|
||
ax.plot(alina_df["Datum"], alina_df["Kopfumfang"].interpolate(limit_area="inside"),color="green", linewidth=1.5, marker="o")
|
||
|
||
last_idx = alina_df["Kopfumfang"].last_valid_index()
|
||
who_row = who_head_daily.loc[who_head_daily["Datum"] == alina_df["Datum"][last_idx]].iloc[0]
|
||
perzentil = (alina_df["Kopfumfang"][last_idx]-who_row["P50"])/((who_row["P97"]-who_row["P50"])/47)+50
|
||
|
||
ax.text(alina_df["Datum"].iloc[-1]+pd.DateOffset(days=1), alina_df["Kopfumfang"].iloc[-1]+0.5,f"{perzentil:.2f} %",
|
||
fontsize=10,
|
||
ha="left",
|
||
va="bottom",
|
||
color="green")
|
||
|
||
ax.set_ylabel("Kopfumfang [cm]")
|
||
# Enable minor ticks
|
||
ax.minorticks_on()
|
||
|
||
# Configure major grid (e.g., solid, darker)
|
||
ax.grid(which='major', color='black', linestyle='-', linewidth=0.8)
|
||
|
||
# Configure minor grid (e.g., dashed, lighter, thinner)
|
||
ax.grid(which='minor', color='gray', linestyle='--', linewidth=0.4, alpha=0.5)
|
||
|
||
# ---------- X‑AXIS ----------
|
||
axes[-1].set_xlabel("Datum")
|
||
|
||
N = 7 # label every Nth tick
|
||
|
||
axes[-1].set_xticks(x[::N])
|
||
axes[-1].set_xticklabels(
|
||
x.iloc[::N].dt.strftime("%d.%m.%Y"),
|
||
rotation=45,
|
||
ha="right"
|
||
)
|
||
|
||
plt.tight_layout()
|
||
plt.savefig(output_path, dpi=150)
|
||
plt.close()
|
||
|
||
if hostname == "serverhost":
|
||
print("Serverhost Mode")
|
||
import subprocess
|
||
|
||
subprocess.run(["bash", "/home/lucas/scripts/serverhost/syncNC.sh", "-al"])
|
||
|
||
print("Fin")
|