用戶自定義JSON格式從熊貓據幀

Question

我有一個熊貓dataFrame.After打印大熊貓數據框的結果看起來像下面

country  branch  no_of_employee  total_salary count_DOB count_email 
    x   a   30     2500000  20   25 
    x   b   20     350000   15   20 
    y   c   30     4500000  30   30 
    z   d   40     5500000  40   40 
    z   e   10     1000000  10   10 
    z   f   15     1500000  15   15 

我想這個轉換成用戶定義的用戶格式像

{ 
     "x": [ 
     { 
      "Branch": "a", 
      "no_employee": 30 
     }, 
     { 
      "Branch": "b", 
      "no_employee": 20 
     } 

     ], 
     "y": [ 
     { 
      "Branch": "c", 
      "no_employee": 30 
     }, 
     { 
      "Branch": "d", 
      "no_employee": 40 
     } 

     ], 
     "z": [ 
     { 
      "Branch": "d", 
      "no_employee": 40 
     }, 
     { 
      "Branch": "e", 
      "no_employee": 10 
     }, 
     { 
      "Branch": "f", 
      "no_employee": 15 
     } 

    ] 

} 

如何將此數據框轉換爲此格式

Answer 1

您可以嘗試groupby與applyto_dict並且最後to_json：

g = df.groupby('country')[["branch", "no_of_employee"]] 
               .apply(lambda x: x.to_dict(orient='records')) 
print g.to_json() 

{ 
    "x": [{ 
     "no_of_employee": 30, 
     "branch": "a" 
    }, { 
     "no_of_employee": 20, 
     "branch": "b" 
    }], 
    "y": [{ 
     "no_of_employee": 30, 
     "branch": "c" 
    }], 
    "z": [{ 
     "no_of_employee": 40, 
     "branch": "d" 
    }, { 
     "no_of_employee": 10, 
     "branch": "e" 
    }, { 
     "no_of_employee": 15, 
     "branch": "f" 
    }] 
}