熊貓：轉換數據框欄與欄標題爲字典鍵和col值字典作爲字典值

Question

我有一個看起來像一個數據幀：

 NAME    ID 
155 ARBITRARY_A 697381 
208 ARBITRARY_B 691820 
265 ARBITRARY_C 691782 
272 ARBITRARY_D 695593 

我想把它轉換成一個字典列表，看起來如：

[{name:ARBITRARY_A, id:697381}, {name:ARBITRARY_B, id:691820}, 
{name:ARBITRARY_C, id:691782}, {name:ARBITRARY_D, id:695593}] 

什麼是最快/最佳的方式來做這個操作？

Answer 1

正如評論指出，pandas.DataFrame.to_dict()可以使用。而在你的情況，你需要orient爲record：

代碼：

df.to_dict('record') 

測試代碼：

df = pd.read_fwf(StringIO(u""" 
      NAME    ID 
    155 ARBITRARY_A 697381 
    208 ARBITRARY_B 691820 
    265 ARBITRARY_C 691782 
    272 ARBITRARY_D 695593"""), 
       header=1, index_col=0) 

print(df) 
print(df.to_dict('record')) 

結果：

  NAME  ID 
155 ARBITRARY_A 697381 
208 ARBITRARY_B 691820 
265 ARBITRARY_C 691782 
272 ARBITRARY_D 695593 

[{u'NAME': u'ARBITRARY_A', u'ID': 697381L}, {u'NAME': u'ARBITRARY_B', u'ID': 691820L}, {u'NAME': u'ARBITRARY_C', u'ID': 691782L}, {u'NAME': u'ARBITRARY_D', u'ID': 695593L}] 

Answer 2

嘗試：

df.to_dict() 

如果這不會產生預期的結果嘗試調換數據框：

df.T.to_dict() 

Answer 3

您可以通過執行推導自己

v = df.values.tolist() 
c = df.columns.values.tolist() 

[dict(zip(c, x)) for x in v] 

[{'ID': 697381L, 'NAME': 'ARBITRARY_A'}, 
{'ID': 691820L, 'NAME': 'ARBITRARY_B'}, 
{'ID': 691782L, 'NAME': 'ARBITRARY_C'}, 
{'ID': 695593L, 'NAME': 'ARBITRARY_D'}] 

小給擠多一點表現出來的df

較大d1