feat: allow multiple keys to be passed to pluck

Suor · Dec 14, 2023 · 93bee01 · 93bee01
1 parent f9e3417
commit 93bee01
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/funcy/colls.py b/funcy/colls.py
@@ -349,8 +349,11 @@ def where(mappings, **cond):
     return filter(match, mappings)
 
 def pluck(key, mappings):
-    """Iterates over values for key in mappings."""
-    return map(itemgetter(key), mappings)
+    """Iterates over values for key, or multiple keys, in mappings."""
+    if isinstance(key, (list, tuple)):
+        return map(itemgetter(*key), mappings)
+    else:
+        return map(itemgetter(key), mappings)
 
 def pluck_attr(attr, objects):
     """Iterates over values of given attribute of given objects."""