熊貓據幀與dict上多列和值列出

Question

我有一個數據幀

id key 
a1  1 
a2  1 
a3  1 
a4  2 
a5  2 
a6  3 

我想創建一個字典，key機器沒有了，id列清單

，如：

{1: ['a1', 'a2', 'a3'], 2: ['a4', 'a5'], 3: ['a6']} 

我可以先使用groupby然後做.to_dict嗎？

Answer 1

我相信你需要列出的dict廣告值 - 使用groupby + apply + to_dict：

d = df.groupby('key')['id'].apply(list).to_dict() 
print (d) 
{1: ['a1', 'a2', 'a3'], 2: ['a4', 'a5'], 3: ['a6']} 

或者，如果需要list與標量增加if/else到apply：

d = df.groupby('key')['id'].apply(lambda x: list(x) if len(x) > 1 else x.iat[0]).to_dict() 
print (d) 
{1: ['a1', 'a2', 'a3'], 2: ['a4', 'a5'], 3: 'a6'} 

Answer 2

使用字典解析圍繞groupby迭代器

{n: v.tolist() for n, v in df.groupby('key').id}