主页 / coding / 问题 / 77977777
Accepted
FreelanceConsultant
Asked: 2024-02-12 01:49:56 +0800 CST

如何在Python中对字典进行分区？

  • 772

许多语言都提供标准库，其中提供字典或二叉树映射类型。许多这些数据结构实现都提供了根据某些条件将数据结构拆分为一对数据结构的功能。

这称为分区操作。Rust 称之为“分裂”。C++map类型具有用于与键进行比较的“下限”和“上限”操作。标准库还提供了存在于std::algorithm.

我看不到在 Python 中执行此操作的理想方法。我设法想出了一种手动方法来做到这一点，但我真的不喜欢这段代码，因为它没有将 adict分成两部分，而是将其中的一部分拆分，然后再将其拆分为另一部分，从而导致脆弱（和重复）的代码。

map = {
   ...
}

map_part_1 = {
    key: value for key, value in map.items() if <condition>
}

map_part_2 = {
    key: value for key, value in map.items() if not <condition>
}

不太好，对吧？有没有更好的办法？

python

3 个回答

  1. Best Answer

    只是一个更短的助手（比您的答案），它仅对每个项目评估一次条件：

    def dict_partition(input: dict, condition: Callable[[Any, Any], bool]) -> tuple[dict, dict]:
    dicts = ({}, {})
    for key, value in input.items():
        dicts[condition(key, value)][key] = value
    return dicts

    如果您想保持第一个字典是真实评估条件的字典，第二个字典是错误评估条件的字典的顺序，请替换return dicts为return dicts[::-1]。

    在线尝试这个！

    • 2

  2. 我不相信这是最好的方法。它可以工作，但需要编写一个辅助函数。

    def dict_partition(input: dict, condition: Callable[[Any, Any], bool]) -> tuple[dict, dict]:

    dict_1 = {
        key: value
            for key, value in input.items()
                if condition(key, value)
    }

    dict_2 = {
        key: value
            for key, value in input.items()
                if not condition(key, value)
    }

    return (dict_1, dict_2)

    然后将其与 a 一起使用lambda，如下所示：

    (dict_1, dict_2) = \
    dict_partition(map, lambda key, value: key < 3)

    这个测试用例证明它是有效的。

    from .dict_partition import dict_partition

def test_dict_partition():

    input = {
        1: 'one',
        2: 'two',
        3: 'three',
        4: 'four',
        5: 'five'
    }

    (dict_1, dict_2) = dict_partition(input, lambda key, value: key < 3)

    dict_1_expected = {
        1: 'one',
        2: 'two'
    }

    dict_2_expected = {
        3: 'three',
        4: 'four',
        5: 'five'
    }

    assert dict_1 == dict_1_expected, 'dict_1 fail test case'
    assert dict_2 == dict_2_expected, 'dict_2 fail test case'

    但是也许有一些更好的解决方案不需要我们编写自己的辅助函数来完成它？

    • 1

  3. 也许我遗漏了一些东西，但你可以在一个循环中完成此操作：

    def dict_partition(dct, condition):
    d1, d2 = {}, {}
    for k, v in dct.items():
        if condition(v):
            d1[k] = v
        else:
            d2[k] = v
    return d1, d2


dct = {"a": 1, "b": 2, "c": 3, "d": 4}
print(dict_partition(dct, lambda x: x > 2))

    印刷：

    ({'c': 3, 'd': 4}, {'a': 1, 'b': 2})

    如果你想修改输入字典，你可以使用dict.pop：

    dct1 = {"a": 1, "b": 2, "c": 3, "d": 4}
dct2 = {k: dct1.pop(k) for k in [*dct1] if dct1[k] > 2}

print(dct1)
print(dct2)

    {'a': 1, 'b': 2}
{'c': 3, 'd': 4}
    • 1

相关问题