How to populate a dictionary of lists using comprehension