fix codes

remo-rcm · Jul 29, 2024 · 566edf9 · 566edf9
1 parent a1a2c49
commit 566edf9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pyremo/codes.py b/pyremo/codes.py
@@ -151,6 +151,8 @@ def _search_df(df, **kwargs):
             cond = "(df['{0}'].isin({1}))".format(key, repr(item))
         elif item is None:
             cond = "(df['{0}'].isna())".format(key)
+        else:
+            cond = "(df['{0}'] == {1})".format(key, repr(item))
         condition_list.append(cond)
     conditions = " & ".join(condition_list)
     return df[eval(conditions)]