dict_of_list.py

from collections import defaultdict

def dict_of_list(keys, values):
    assert(len(keys) == len(values))
    key2values = defaultdict(list)
    for k, v in zip(keys, values):
        key2values[k].append(v)
    return key2values