每个键具有多个值的字典 python

Question

我有一个 dataframe -

ID T Q1 P Q2
10 xy 1 pq 2
20 yz 1 rs 1
20 ab 1 tu 2
30 cd 2 cu 2
30 xy 1 mu 1
30 bb 1 bc 1

现在我需要一个以 Id 为键的字典，并将列值的 rest 作为列表作为字典的值

output ：

{10:['xy',1,'pq',2]}
{20:['ab',1,'tu',2]}
{30:['bb',1,'bc',1]}

预期结果：

{10:[['xy',1,'pq',2]]}
{20:[['yz',1,'rs',1],['ab',1,'tu',2]]}
{30:[['cd',2,'cu',2],['xy',1,'mu',1],['bb',1,'bc',1]]}

Answer 1

尝试：

x = (
    df.groupby("ID")
    .apply(lambda x: x.iloc[:, 1:].agg(list).values.tolist())
    .to_dict()
)
print(x)

印刷：

{10: [['xy', 1, 'pq', 2]], 
 20: [['yz', 1, 'rs', 1], ['ab', 1, 'tu', 2]], 
 30: [['cd', 2, 'cu', 2], ['xy', 1, 'mu', 1], ['bb', 1, 'bc', 1]]}