[variability] nulls in groupings are equal

This commit is contained in:
jan 2025-12-30 02:44:40 -08:00
parent 47d173443f
commit 4ba0eb09e8

View File

@ -67,7 +67,7 @@ def variability_plot(
# Assign category indicies (x_pos)
df = df.sort(groups)
df_groups = df.select(groups).unique(maintain_order=True).with_row_index(name='x_pos')
df = df.join(df_groups, on=groups, maintain_order='left')
df = df.join(df_groups, on=groups, maintain_order='left', nulls_equal=True)
max_group_length = df.group_by(groups).len().select('len').max()[0, 0] # How many points in the largest x_pos
# Add jitter to the scatterplots-plots