utils.py 287 Bytes
Newer Older
1
2
3
4
5
6
7
8
def get_data_per_category(data, categories):
    data_per_category = {category: [] for category in categories}
    for item in data:
        category = item["category"]
        if category in categories:
            data_per_category[category].append(item)

    return data_per_category