[英]Convert a column in DataFrame to list and another column to set
我想将 dataframe 中的一些列转换为列表,将一些列转换为具有相似名称的设置给定的 dataframe,df:
Name id other list
0 ben 00005 abc [1000, A, 90]
1 alex 00006 gf [3000, B, 80]
2 linn 00007 jgj [600, C, 55]
3 luke 00009 gg [5000, D, 88]
4 alex 00001 gf [7000, R, 98]
5 ben 00002 abc [9000, S, 28]
6 ben 00003 abc [5000, T, 48]
所需的 output,df1:
Name id other list
0 ben {00005, 0002,0003} abc [[1000, A, 90],[9000, S, 28],[5000, T, 48]]
1 alex {00006,0001} gf [3000, B, 80], [7000, R, 98]
2 linn {00007} jgj [600, C, 55]
3 luke {00009} gg [5000, D, 88]
您可以将.groupby()
与.agg()
) 一起使用:
df.groupby("Name", as_index=False).agg({"id": set, "other": "first", "list": list})
这输出:
Name id other list
0 alex {00001, 00006} gf [[3000, B, 80], [7000, R, 98]]
1 ben {00005, 00003, 00002} abc [[1000, A, 90], [9000, S, 28], [5000, T, 48]]
2 linn {00007} jgj [[600, C, 55]]
3 luke {00009} gg [[5000, D, 88]]
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.