import pandas as pd def contains_code(crs_codes, code_list): codes = str(crs_codes).split(';') return any(code in code_list for code in codes) def filter_projects(df, crs3_list, crs5_list, sdg_str): if crs3_list != [] or crs5_list != [] or sdg_str != "": if crs3_list and not crs5_list: df = df[df['crs_3_code'].apply(lambda x: contains_code(x, crs3_list))] elif crs3_list and crs5_list: df = df[df['crs_5_code'].apply(lambda x: contains_code(x, crs5_list))] elif not crs3_list and crs5_list: df = df[df['crs_5_code'].apply(lambda x: contains_code(x, crs5_list))] if sdg_str != "": df = df[df["sgd_pred_code"] == int(sdg_str)] return df