python 数据分类

#!/usr/bin/env python
# coding: utf-8

import six

def sort_items_by_key(items, key):
    result = {}
    for k, v in key.items():
        if isinstance(v, (six.string_types, six.integer_types)):
            tmp = filter(lambda x: x.get(k) == v, items)
            result.setdefault(v, []).extend(tmp)
        elif isinstance(v, (list, set, tuple)):
            for vv in v:
                tmp = filter(lambda x: x.get(k) == vv, items)
                result.setdefault(vv, []).extend(tmp)
    return result


def test_items_key():
    items = [
        {"id": 1, "name": "atom", "age": 25},
        {"id": 2, "name": "shylock", "age": 26},
        {"id": 3, "name": "john", "age": 26}
    ]

    key = {"id": [1,2,3], "age": 26}

    print(sort_items_by_key(items, key))

if __name__ == "__main__":
    test_items_key()

output：
    {1: [{‘age‘: 25, ‘id‘: 1, ‘name‘: ‘atom‘}], 26: [{‘age‘: 26, ‘id‘: 2, ‘name‘: ‘shylock‘}, {‘age‘: 26, ‘id‘: 3, ‘name‘: ‘john‘}], 3: [{‘age‘: 26, ‘id‘: 3, ‘name‘: ‘john‘}], 2: [{‘age‘: 26, ‘id‘: 2, ‘name‘: ‘shylock‘}]}