lkj 0.1.39__py3-none-any.whl → 0.1.41__py3-none-any.whl

lkj/__init__.py

@@ -13,6 +13,7 @@ from lkj.dicts import (
     inclusive_subdict,  # new dictionary with only the keys in `include`
     exclusive_subdict,  # new dictionary with only the keys not in `exclude`.
     merge_dicts,  # Merge multiple dictionaries recursively
+    compare_field_values,  # Compare two dictionaries' values
 )
 from lkj.filesys import get_app_data_dir, get_watermarked_dir, enable_sourcing_from_file
 from lkj.strings import (

lkj/dicts.py

@@ -111,16 +111,10 @@ from typing import Mapping, Callable, TypeVar, Iterable, Tuple
 KT = TypeVar("KT")  # Key type
 VT = TypeVar("VT")  # Value type
 
+
 # Note: Could have all function parameters (recursive_condition, etc.) also take the
 #       enumerated index of the mapping as an argument. That would give us even more
 #       flexibility, but it might be overkill and make the interface more complex.
-from typing import Mapping, Callable, TypeVar, Iterable, Tuple
-from collections import defaultdict
-
-KT = TypeVar("KT")  # Key type
-VT = TypeVar("VT")  # Value type
-
-
 def merge_dicts(
     *mappings: Mapping[KT, VT],
     recursive_condition: Callable[[VT], bool] = lambda v: isinstance(v, Mapping),
@@ -221,3 +215,62 @@ def merge_dicts(
                 merged[key] = value
 
     return merged
+
+
+import operator
+from typing import Callable, Dict, Any
+
+Comparison = Any
+Comparator = Callable[[dict, dict], Comparison]
+
+
+def compare_field_values(
+    dict1,
+    dict2,
+    *,
+    field_comparators: Dict[KT, Comparator] = {},
+    default_comparator: Comparator = operator.eq,
+    aggregator: Callable[
+        [Dict[KT, Comparison]], Any
+    ] = lambda d: d,  # lambda d: np.mean(list(d.values()))
+):
+    """
+    Compare two dictionaries' values field by field
+
+    :param dict1: The first dictionary.
+    :param dict2: The second dictionary.
+    :param field_comparators: A dictionary where keys are field names and values are comparator functions.
+    :param default_comparator: A default comparator function to use if no specific comparator is provided for a field.
+    :param aggregator: A function to aggregate the comparison results into a final comparison object.
+    :return: A final score based on the comparison results.
+
+    >>> dict1 = {"color": "brown", "animal": "dog"}
+    >>> dict2 = {"color": "brown", "animal": "cat"}
+    >>> dict3 = {"color": "brown", "animal": "bird"}
+    >>> field_comparators = {
+    ...     "color": lambda x, y: 1 if x == y else 0,
+    ...     "animal": lambda x, y: 1 if len(x) == len(y) else 0
+    ... }
+    >>> compare_field_values(dict1, dict2, field_comparators=field_comparators)
+    {'color': 1, 'animal': 1}
+    >>> compare_field_values(dict1, dict3, field_comparators=field_comparators)
+    {'color': 1, 'animal': 0}
+    >>> import functools, statistics
+    >>> aggregator = lambda d: statistics.mean(d.values())
+    >>> mean_of_values = functools.partial(
+    ...     compare_field_values, field_comparators=field_comparators, aggregator=aggregator
+    ... )
+    >>> mean_of_values(dict1, dict2)
+    1
+    >>> mean_of_values(dict1, dict3)
+    0.5
+
+    """
+    common_keys = [k for k in dict1.keys() if k in dict2.keys()]
+
+    comparisons = {}
+    for key in common_keys:
+        comparator = field_comparators.get(key, default_comparator)
+        comparisons[key] = comparator(dict1[key], dict2[key])
+
+    return aggregator(comparisons)

{lkj-0.1.39.dist-info → lkj-0.1.41.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lkj
-Version: 0.1.39
+Version: 0.1.41
 Summary: A dump of homeless useful utils
 Home-page: https://github.com/thorwhalen/lkj
 Author: Thor Whalen

{lkj-0.1.39.dist-info → lkj-0.1.41.dist-info}/RECORD

@@ -1,6 +1,6 @@
-lkj/__init__.py,sha256=YEPCVj-0EJOfpapgX3Vw07399w13OkCm39KCmCi5AQg,6806
+lkj/__init__.py,sha256=Vceal0uY2jW1y0sQHOk1WHXc9aTAEXXOJ9vgyOmnrNk,6868
 lkj/chunking.py,sha256=RpNdx5jEuO4mFg2qoTkD47iL35neBneuZ5xgQ_cBkiM,3755
-lkj/dicts.py,sha256=EbTLtYlexCDhDlwKnzflZ5CEU0ql6KgZdu3SvR7bU20,8289
+lkj/dicts.py,sha256=vmdOXgGVWe3LTz40dHamj7-c8ob4B5SxL3zPJcylf40,10216
 lkj/filesys.py,sha256=NbWDuc848h8O42gwX7d9yNJkrWBgzSFnkoEdSRgvBAg,8883
 lkj/funcs.py,sha256=LXJlj3gMMsbD0t2gn2NZZ6mOqmW5bxM-94uGoYgrhzI,8930
 lkj/importing.py,sha256=_BFBmdaKCBnk5broltWDeAPeEKG1dEkS8DhIdv6UhSI,3809
@@ -8,8 +8,8 @@ lkj/iterables.py,sha256=9jeO36w-IGiZryge7JKgXZOGZAgehUvhwKV3nHPcZWk,2801
 lkj/loggers.py,sha256=ImmBdacz89Lvb3dg_xI5jOct_44rSRv0hNI_CVehy60,13706
 lkj/misc.py,sha256=IZf05tkl0cgiMgBwCMH5cLSC59fRXwnemPRo8G0OxQg,1436
 lkj/strings.py,sha256=3YjlxOWUfzWqwu51uq_pv9XZReLqRFvziGtRsdzEtw8,24662
-lkj-0.1.39.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-lkj-0.1.39.dist-info/METADATA,sha256=FFVXFg-ElX-E_xQRF9i8IExOOqEgIScsqigQTu0vtdM,4684
-lkj-0.1.39.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-lkj-0.1.39.dist-info/top_level.txt,sha256=3DZOUwYmyurJFBXQCvCmEIVm8z2b42O5Sx3RDQyePfg,4
-lkj-0.1.39.dist-info/RECORD,,
+lkj-0.1.41.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+lkj-0.1.41.dist-info/METADATA,sha256=Tc9RoFOGPs137L9Mw1--sGkVEtmc455ZLx2lHRMqUDc,4684
+lkj-0.1.41.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+lkj-0.1.41.dist-info/top_level.txt,sha256=3DZOUwYmyurJFBXQCvCmEIVm8z2b42O5Sx3RDQyePfg,4
+lkj-0.1.41.dist-info/RECORD,,