initialize new marks with 0 for new samples in bar plots

Chilipp · Feb 19, 2019 · 2a080c7 · 2a080c7
1 parent 1d79748
commit 2a080c7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/straditize/straditizer.py b/straditize/straditizer.py
@@ -1202,7 +1202,8 @@ def _new_mark(pos, artists=[]):
 
         def new_mark(pos):
             return [_new_mark(
-                [np.array(starts + full_df.loc[np.round(pos[-1] - ylim[0])]),
+                [np.array(starts +
+                          full_df.loc[np.round(pos[-1] - ylim[0])].fillna(0)),
                  np.round(pos[-1])])]
 
         def new_mark_and_range(key, row, row_indices):