[fix] replace NaN to [np.nan]

KamitaniLab · Oct 9, 2024 · 56e76b4 · 56e76b4
1 parent d7e5c18
commit 56e76b4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/bdpy/fig/makeplots2.py b/bdpy/fig/makeplots2.py
@@ -593,6 +593,10 @@ def __weird_form_to_long(df, target_col, identify_cols=[]):
     identify_cols: list of str
         A list of columns to keep in the decomposed dataframe, other than target_col.
     """
+    # replace NaN to [np.nan]
+    nans = df[target_col].isna()
+    df[target_col][nans] = [np.nan]
+    # Weird form
     df_result = pd.DataFrame()
     for i, row in df.iterrows():
         tmp = {}