perf(rust, python): use hash set in drop_many (#9807)

pola-rs · Jul 11, 2023 · 243951c · 243951c
1 parent f5f0630
commit 243951c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/polars/polars-core/src/frame/mod.rs b/polars/polars-core/src/frame/mod.rs
@@ -1074,8 +1074,8 @@ impl DataFrame {
     }
 
     pub fn drop_many<S: AsRef<str>>(&self, names: &[S]) -> Self {
-        let names = names.iter().map(|s| s.as_ref()).collect();
-        fn inner(df: &DataFrame, names: Vec<&str>) -> DataFrame {
+        let names: PlHashSet<_> = names.iter().map(|s| s.as_ref()).collect();
+        fn inner(df: &DataFrame, names: PlHashSet<&str>) -> DataFrame {
             let mut new_cols = Vec::with_capacity(df.columns.len() - names.len());
             df.columns.iter().for_each(|s| {
                 if !names.contains(&s.name()) {