創建基於熊貓據幀

Question

嵌套的字典我有這樣

data = np.zeros((3,), dtype=[('A', 'a10'), ('B', 'f4'), ('C', 'f4')]) 
data[:] = [('key1', 1, 2), ('key2', 3, 4), ('key3', 5, 6)] 
df = pd.DataFrame(data) 

一個數據幀，我想將其轉換爲dataframes其中的第一列是字典的鍵的字典。我如何創建這樣的字典？

Answer 1

試試這個：

data = np.zeros((3,), dtype=[('A', 'a10'), ('B', 'f4'), ('C', 'f4')]) 
data[:] = [('key1', 1, 2), ('key2', 3, 4), ('key3', 5, 6)] 
df = pd.DataFrame(data) 

df：

  A B C 
0 b'key1' 1 2 
1 b'key2' 3 4 
2 b'key3' 5 6 

轉換：

df.set_index('A').T.to_dict() 

結果：

{b'key1': {'B': 1.0, 'C': 2.0}, 
b'key2': {'B': 3.0, 'C': 4.0}, 
b'key3': {'B': 5.0, 'C': 6.0}}