Pandas dataframe to dictionary with specific format

user_14277243 0
Viewed 42

I have a df like this:

Customer#   Facility   Transp
1           RS         4
2           RS         7
3           RS         9
1           CM         2
2           CM         8
3           CM         5

I want to convert to a dictionary that looks like this:
transp = {'RS' : {1 : 4, 2 : 7, 3 : 9, 
         'CM' : {1 : 2, 2 : 8, 3 : 5}}

I'm unfamiliar with this conversion. I tried various options. The data has to be exactly in this dictionary format. I can't have nesting with []. Essentially Facility is the primary level then Customer / Transp. I feel like this should be easy.... Thanks,

pandas dataframe dictionary nested python
1 Answers

You can do it in one go.

df = pd.DataFrame({"Customer#": [1, 2, 3, 1, 2, 3],
                   "Facility": ['RS', 'RS', 'RS', 'CM', 'CM', 'CM'],
                   "Transp": [4, 7, 9, 2, 8, 5]})

transp = df.groupby('Facility')[['Customer#','Transp']].apply(lambda g: dict(g.values.tolist())).to_dict()

print(transp)
user_64183270
Related