Fix plot_frequencies_time_series reading wrong cohort after filtering (fixes #827)

blankkiri · blankkiri · commit feaee7ae2f2f · 2026-02-20T17:26:51.000+05:30
diff --git a/malariagen_data/anoph/frq_base.py b/malariagen_data/anoph/frq_base.py
@@ -386,7 +386,7 @@ def plot_frequencies_time_series(
         # Build a long-form dataframe from the dataset.
         dfs = []
         for cohort_index, cohort in enumerate(df_cohorts.itertuples()):
-            ds_cohort = ds.isel(cohorts=cohort_index)
+            ds_cohort = ds.isel(cohorts=cohort.Index)
             df = pd.DataFrame(
                 {
                     "taxon": cohort.taxon,

Original file line number	Diff line number	Diff line change
`@@ -386,7 +386,7 @@ def plot_frequencies_time_series(`
`386`	`386`	`# Build a long-form dataframe from the dataset.`
`387`	`387`	`dfs = []`
`388`	`388`	`for cohort_index, cohort in enumerate(df_cohorts.itertuples()):`
`389`		`- ds_cohort = ds.isel(cohorts=cohort_index)`
	`389`	`+ ds_cohort = ds.isel(cohorts=cohort.Index)`
`390`	`390`	`df = pd.DataFrame(`
`391`	`391`	`{`
`392`	`392`	`"taxon": cohort.taxon,`