轉換大熊貓據幀與dict其中每個值是多個列的值的列表

Question

比方說，我有數據幀

filename size inverse similarity 
123.txt 1 2  34 
323.txt 3 1  44 
222.txt 4 1  43 

我想創建一個字典的形式

{'123.txt': [1, 2, 34], 
'323.txt': [3, 1, 44], 
'222.txt': [4, 1, 43]} 

解決方案我發現處理的情況下創建單個值的字典使用類似的東西

df.set_index('Filename')['size'].to_dict() 

Answer 1

設置'filename'爲指標，採取轉置，然後使用to_dict與orient='list'：

my_dict = df.set_index('filename').T.to_dict(orient='list') 

輸出結果：

{'323.txt': [3, 1, 44], '222.txt': [4, 1, 43], '123.txt': [1, 2, 34]} 

Answer 2

下面是用字典理解的方法：

{k: v for k, v in zip(df['filename'], df.set_index('filename').values.tolist())} 
Out: {'123.txt': [1, 2, 34], '222.txt': [4, 1, 43], '323.txt': [3, 1, 44]}