fixes #665

Author: jph00

fastcore/_modidx.py

@@ -179,6 +179,7 @@
                                  'fastcore.basics.otherwise': ('basics.html#otherwise', 'fastcore/basics.py'),
                                  'fastcore.basics.partialler': ('basics.html#partialler', 'fastcore/basics.py'),
                                  'fastcore.basics.partition': ('basics.html#partition', 'fastcore/basics.py'),
+                                 'fastcore.basics.partition_dict': ('basics.html#partition_dict', 'fastcore/basics.py'),
                                  'fastcore.basics.patch': ('basics.html#patch', 'fastcore/basics.py'),
                                  'fastcore.basics.patch_property': ('basics.html#patch_property', 'fastcore/basics.py'),
                                  'fastcore.basics.patch_to': ('basics.html#patch_to', 'fastcore/basics.py'),

fastcore/basics.py

@@ -11,16 +11,16 @@
            'annotations', 'anno_ret', 'signature_ex', 'union2tuple', 'argnames', 'with_cast', 'store_attr', 'attrdict',
            'properties', 'camel2words', 'camel2snake', 'snake2camel', 'class2attr', 'getcallable', 'getattrs',
            'hasattrs', 'setattrs', 'try_attrs', 'GetAttrBase', 'GetAttr', 'delegate_attr', 'ShowPrint', 'Int', 'Str',
-           'Float', 'partition', 'flatten', 'concat', 'strcat', 'detuplify', 'replicate', 'setify', 'merge', 'range_of',
-           'groupby', 'last_index', 'filter_dict', 'filter_keys', 'filter_values', 'cycle', 'zip_cycle', 'sorted_ex',
-           'not_', 'argwhere', 'filter_ex', 'renumerate', 'first', 'last', 'only', 'nested_attr', 'nested_setdefault',
-           'nested_callable', 'nested_idx', 'set_nested_idx', 'val2idx', 'uniqueify', 'loop_first_last', 'loop_first',
-           'loop_last', 'first_match', 'last_match', 'fastuple', 'bind', 'mapt', 'map_ex', 'compose', 'maps',
-           'partialler', 'instantiate', 'using_attr', 'copy_func', 'patch_to', 'patch', 'patch_property', 'compile_re',
-           'ImportEnum', 'StrEnum', 'str_enum', 'ValEnum', 'Stateful', 'NotStr', 'PrettyString', 'even_mults',
-           'num_cpus', 'add_props', 'str2bool', 'str2int', 'str2float', 'str2list', 'str2date', 'to_bool', 'to_int',
-           'to_float', 'to_list', 'to_date', 'typed', 'exec_new', 'exec_import', 'lt', 'gt', 'le', 'ge', 'eq', 'ne',
-           'add', 'sub', 'mul', 'truediv', 'is_', 'is_not', 'mod']
+           'Float', 'partition', 'partition_dict', 'flatten', 'concat', 'strcat', 'detuplify', 'replicate', 'setify',
+           'merge', 'range_of', 'groupby', 'last_index', 'filter_dict', 'filter_keys', 'filter_values', 'cycle',
+           'zip_cycle', 'sorted_ex', 'not_', 'argwhere', 'filter_ex', 'renumerate', 'first', 'last', 'only',
+           'nested_attr', 'nested_setdefault', 'nested_callable', 'nested_idx', 'set_nested_idx', 'val2idx',
+           'uniqueify', 'loop_first_last', 'loop_first', 'loop_last', 'first_match', 'last_match', 'fastuple', 'bind',
+           'mapt', 'map_ex', 'compose', 'maps', 'partialler', 'instantiate', 'using_attr', 'copy_func', 'patch_to',
+           'patch', 'patch_property', 'compile_re', 'ImportEnum', 'StrEnum', 'str_enum', 'ValEnum', 'Stateful',
+           'NotStr', 'PrettyString', 'even_mults', 'num_cpus', 'add_props', 'str2bool', 'str2int', 'str2float',
+           'str2list', 'str2date', 'to_bool', 'to_int', 'to_float', 'to_list', 'to_date', 'typed', 'exec_new',
+           'exec_import', 'lt', 'gt', 'le', 'ge', 'eq', 'ne', 'add', 'sub', 'mul', 'truediv', 'is_', 'is_not', 'mod']
 
 # %% ../nbs/01_basics.ipynb
 from .imports import *
@@ -574,6 +574,13 @@ def partition(coll, f):
         ts,fs = typ(ts),typ(fs)
     return ts,fs
 
+# %% ../nbs/01_basics.ipynb
+def partition_dict(d, f):
+    "Partition a dict by a predicate that takes key/value params"
+    ts,fs = {},{}
+    for k,v in d.items(): (fs,ts)[f(k,v)][k] = v
+    return ts,fs
+
 # %% ../nbs/01_basics.ipynb
 def flatten(o):
     "Concatenate all collections and items as a generator"

nbs/01_basics.ipynb

@@ -672,14 +672,6 @@
    "execution_count": null,
    "metadata": {},
    "outputs": [
-    {
-     "name": "stderr",
-     "output_type": "stream",
-     "text": [
-      "/Users/jhoward/miniforge3/lib/python3.12/site-packages/tqdm/auto.py:21: TqdmWarning: IProgress not found. Please update jupyter and ipywidgets. See https://ipywidgets.readthedocs.io/en/stable/user_install.html\n",
-      "  from .autonotebook import tqdm as notebook_tqdm\n"
-     ]
-    },
     {
      "data": {
       "text/markdown": [
@@ -3388,6 +3380,43 @@
     "test_eq(ts, [1,3,5,7,9])"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "#| export\n",
+    "def partition_dict(d, f):\n",
+    "    \"Partition a dict by a predicate that takes key/value params\"\n",
+    "    ts,fs = {},{}\n",
+    "    for k,v in d.items(): (fs,ts)[f(k,v)][k] = v\n",
+    "    return ts,fs"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "d = {'a':1, 'b':2, 'c':3, 'd':4}\n",
+    "ts,fs = partition_dict(d, lambda k,v: v%2)\n",
+    "test_eq(fs, {'b':2, 'd':4})\n",
+    "test_eq(ts, {'a':1, 'c':3})"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "ts,fs = partition_dict(d, lambda k,v: k in 'bc')\n",
+    "test_eq(ts, {'b':2, 'c':3})\n",
+    "test_eq(fs, {'a':1, 'd':4})"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,