Avoid using pd.unique()

IAMconsortium · Sep 1, 2023 · 652ee27 · 652ee27
1 parent b16ae63
commit 652ee27
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/pyam/utils.py b/pyam/utils.py
@@ -513,10 +513,10 @@ def _get_col(c):
         except KeyError:
             return df[c]
 
-    index = list(zip(*[_get_col(col) for col in cols]))
-    if unique:
-        index = pd.unique(index)
-    return pd.MultiIndex.from_tuples(index, names=tuple(cols))
+    index = pd.MultiIndex.from_tuples(
+        list(zip(*[_get_col(col) for col in cols])), names=tuple(cols)
+    )
+    return index.drop_duplicates() if unique else index
 
 
 def pattern_match(