Fixes #392.

Author: Jacobsen

toolz/dicttoolz.py

@@ -313,3 +313,21 @@ def get_in(keys, coll, default=None, no_default=False):
         if no_default:
             raise
         return default
+
+
+def select_keys(d, keys, factory=dict):
+    """ Select only certain keys from a dictionary
+
+    If supplied keys are not found in the dictionary, returns an empty
+    dictionary.
+
+    >>> d = {'a': 1, 'b': 2, 'c': 3}
+    >>> select_keys(d, ['a', 'b', 'd'])
+    {'a': 1, 'b': 2}
+    >>> select_keys(d, ['c'])
+    {}
+    """
+    rv = factory()
+    for k in set(keys).intersection(d.keys()):
+        rv[k] = d[k]
+    return rv

toolz/tests/test_dicttoolz.py

@@ -1,7 +1,8 @@
 from collections import defaultdict as _defaultdict
-from toolz.dicttoolz import (merge, merge_with, valmap, keymap, update_in,
-                             assoc, dissoc, keyfilter, valfilter, itemmap,
-                             itemfilter, assoc_in)
+from toolz.dicttoolz import (assoc, assoc_in, dissoc, itemfilter,
+                             itemmap, keyfilter, keymap, merge,
+                             merge_with, select_keys, update_in,
+                             valfilter, valmap)
 from toolz.utils import raises
 from toolz.compatibility import PY3
 
@@ -117,6 +118,14 @@ def test_assoc_in(self):
         assert d is oldd
         assert d2 is not oldd
 
+    def test_select_keys(self):
+        D, kw = self.D, self.kw
+        assert select_keys(D({}), [], **kw) == D({})
+        assert select_keys(D({"a": 1}), ["a"], **kw) == D({"a": 1})
+        assert select_keys(D({"a": 1}), [], **kw) == D({})
+        assert select_keys(D({"a": 1}), ["b"], **kw) == D({})
+        assert select_keys(D({"a": 1, "b": 2}), ["b"], **kw) == D({"b": 2})
+
     def test_update_in(self):
         D, kw = self.D, self.kw
         assert update_in(D({"a": 0}), ["a"], inc, **kw) == D({"a": 1})