Skip to content

GitLab

Commit 41b01bb4 authored by JOSSOUD Olivier's avatar JOSSOUD Olivier
Browse files

Flow. Correct bug which removed the first rows before first iceblock.

parent c6b67c93
Pipeline #172771 failed with stage
in 54 seconds
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
cfatools/processor/flow.py
View file @ 41b01bb4
......@@ -449,6 +449,9 @@ def add_melted_height(arrival_df: pd.DataFrame, encoder_df: pd.DataFrame, iceblo
df["melted_height_" + suffix + "_icbk"] = df["melted_height_" + suffix] - df.iloc[0]["melted_height_" + suffix]
return df
# Set initial icbk_code to 0 to avoid removal of the first rows
arrival_df.loc[arrival_df["icbk_code"].isna(), "icbk_code"] = 0
arrival_df = arrival_df.groupby("icbk_code").apply(_get_melted_height_by_iceblock_, ("interpstack"))
arrival_df = arrival_df.groupby("icbk_code").apply(_get_melted_height_by_iceblock_, ("expected"))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment