Green_Energy / modules /filters.py
dschandra's picture
Create filters.py
c057d2b verified
raw
history blame
609 Bytes
def apply_filters(df, alert_filter, cam_filter):
"""
Apply filters based on alert level and camera status.
:param df: DataFrame containing the pole data.
:param alert_filter: List of selected alert levels (Green, Yellow, Red).
:param cam_filter: Selected camera status ("All", "Online", "Offline").
:return: Filtered DataFrame.
"""
# Filter by alert level
filtered_df = df[df["Alert Level"].isin(alert_filter)]
# Filter by camera status
if cam_filter != "All":
filtered_df = filtered_df[filtered_df["Camera Status"] == cam_filter]
return filtered_df