CODEWITHSUNDEEP
pythonpandasdataframe
Ujjwal Roy
Ujjwal Roy

Reputation: 73

Convert a pandas dataframe into a dictionary of dictionaries

I have a data frame in which I have attributes and values for two products. 

PRODUCT ATTRIBUTE   VALUES
prod1   Attr1         A
prod1   Attr2         B
prod1   Attr3         C
prod1   Attr4         D
prod2   Attr1         E
prod2   Attr2         F
prod2   Attr3         G
prod2   Attr4         H

How can I convert it into a dictionary of a list of dictionaries that looks like the following?

{'prod1':[{'Attr1':A, 'Attr2':B, 'Attr3':C, 'Attr4':D}], 'prod2':[{'Attr1':E, 'Attr2':F, 'Attr3':G, 'Attr4':H}]}

Upvotes: 2

Views: 4107

Answers (4)

piRSquared
piRSquared

Reputation: 294218

comprehension

{k: [v.to_dict()] for k, v in df.set_index(['PRODUCT', 'ATTRIBUTE']).VALUES.unstack(0).iteritems()}

{'prod1': [{'Attr1': 'A', 'Attr2': 'B', 'Attr3': 'C', 'Attr4': 'D'}],
 'prod2': [{'Attr1': 'E', 'Attr2': 'F', 'Attr3': 'G', 'Attr4': 'H'}]}

Upvotes: 2

roman
roman

Reputation: 117345

You can use pandas.DataFrame.pivot to prepare your data and then call pandas.DataFrame.to_dict:

>>> df.pivot(columns='PRODUCT',index='ATTRIBUTE', values='VALUES').to_dict()
{'prod1': {'Attr4': 'D', 'Attr2': 'B', 'Attr3': 'C', 'Attr1': 'A'}, 'prod2': {'Attr4': 'H', 'Attr2': 'F', 'Attr3': 'G', 'Attr1': 'E'}}

I also assume that you don't actually need one-element lists there. In this case you can have really simple solution. If you need those lists then it's probably better to go with @jezrael answer

Upvotes: 2

Ted Petrou
Ted Petrou

Reputation: 61947

Pivot and use the to_dict method and then wrap the inner dict in a list.

d= df.pivot(index='PRODUCT', columns='ATTRIBUTE', values='VALUES').to_dict()
{key:[value] for key,value in d.items()}

Upvotes: 0

jezrael
jezrael

Reputation: 862511

You can use groupby with apply:

d = df.groupby('PRODUCT').apply(lambda x: [dict(zip(x.ATTRIBUTE, x.VALUES))]).to_dict()
print (d)
{'prod1': [{'Attr1': 'A', 'Attr2': 'B', 'Attr3': 'C', 'Attr4': 'D'}],
'prod2': [{'Attr1': 'E', 'Attr2': 'F', 'Attr3': 'G', 'Attr4': 'H'}]}

Upvotes: 5

Related Questions