CODEWITHSUNDEEP
jsonpandasdataframe
tensor
tensor

Reputation: 3340

How to export pandas dataframe to json in specific format

My dataframe is

  'col1' , 'col2'
    A    ,   89
    A    ,   232
    C    ,   545
    D    ,   998

and would like to export as follow :

{
  'A' : [ 89, 232 ],
  'C' : [545],
  'D' : [998]   
}

However, all the to_json does not fit this format (orient='records', ...). Is there a way to ouput like this ?

Upvotes: 1

Views: 203

Answers (1)

jezrael
jezrael

Reputation: 862611

Use groupby for convert to list and then to_json:

json = df.groupby('col1')['col2'].apply(list).to_json()
print (json)
{"A":[89,232],"C":[545],"D":[998]}

Detail:

print (df.groupby('col1')['col2'].apply(list))
col1
A    [89, 232]
C        [545]
D        [998]
Name: col2, dtype: object

Upvotes: 2

Related Questions