GenAICoder commited on
Commit
fb2d47c
·
verified ·
1 Parent(s): 86aec3b

Create filter_helpers.py

Browse files
Files changed (1) hide show
  1. helper/filter_helpers.py +31 -0
helper/filter_helpers.py ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # helpers/filter_helpers.py
2
+
3
+ def apply_filters(
4
+ df,
5
+ filters=None
6
+ ):
7
+ """
8
+ Generic filtering helper.
9
+
10
+ Example filters:
11
+ {
12
+ "sourcing_channel": ["Digital"],
13
+ "fico_band": ["651-700", "701-750"]
14
+ }
15
+ """
16
+
17
+ if filters is None:
18
+ return df
19
+
20
+ filtered_df = df.copy()
21
+
22
+ for col, values in filters.items():
23
+
24
+ if not isinstance(values, list):
25
+ values = [values]
26
+
27
+ filtered_df = filtered_df[
28
+ filtered_df[col].isin(values)
29
+ ]
30
+
31
+ return filtered_df