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

1import pandas as pd 

2 

3 

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

5 """ 

6 Filters samples to exclude given ids. 

7 

8 Parameters 

9 ---------- 

10 samples 

11 Sample cell ids and coordinates. 

12 exclude 

13 List of ids to exclude. 

14 

15 Returns 

16 ------- 

17 : 

18 Samples without excluded ids. 

19 """ 

20 

21 samples = samples[~samples.id.isin(exclude)] 

22 return samples.reset_index(drop=True)