数据框转换为字典

Question

数据框转换为字典

4

I have following Dataframe:

     image          product  vp_fk               mask
0    136524         105       2316               51322
1    136524         105       2316               51324
2    136524         106       2316               51325
3    136524         106       2316               51328
4    136524         106       2316               51329
5    136525         108       2319               51330

我希望将其转换为以下字典：

{105:[[136524, 2316, 51322], 
      [136524, 2316, 51324]]
106: [[136524, 2316, 51325], 
      [136524, 2316, 51328],
      [136524, 2316, 51329]]
108: [[136525, 2319, 51330]]}

- Night Walker

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- MaxU - stand with Ukraine · Accepted Answer

试一下这个：

In [187]: df.groupby('product')[['image','vp_fk','mask']].apply(lambda grp: grp.values.tolist()).to_dict()
Out[187]:
{105: [[136524, 2316, 51322], [136524, 2316, 51324]],
 106: [[136524, 2316, 51325], [136524, 2316, 51328], [136524, 2316, 51329]],
 108: [[136525, 2319, 51330]]}