Coverage for src/abm_initialization_collection/sample/include_selected_ids.py: 100%

1import pandas as pd 

2 

3 

4def include_selected_ids(samples: pd.DataFrame, include: list[int]) -> pd.DataFrame: 

5 """ 

6 Filters samples to include given ids. 

7 

8 Parameters 

9 ---------- 

10 samples 

11 Sample cell ids and coordinates. 

12 include 

13 List of ids to include. 

14 

15 Returns 

16 ------- 

17 : 

18 Samples with included ids. 

19 """ 

20 

21 samples = samples[samples.id.isin(include)] 

22 return samples.reset_index(drop=True)