functional

Author: molpopgen

src/node_table.rs

@@ -419,13 +419,10 @@ impl NodeTable {
         &self,
         mut f: impl FnMut(&crate::NodeTableRow) -> bool,
     ) -> Vec<NodeId> {
-        let mut samples: Vec<NodeId> = vec![];
-        for row in self.iter() {
-            if f(&row) {
-                samples.push(row.id);
-            }
-        }
-        samples
+        self.iter()
+            .filter(|row| f(row))
+            .map(|row| row.id)
+            .collect::<Vec<_>>()
     }
 }