Skip to main content
 首页 » 编程设计

python之同时按 2 列分组,同时向 groupby 添加一些条件

2025年05月04日22sxdcgaq8080

我想按集合和零件列对我的数据进行分组。如果它们有相同的部分,则将它们组合在一起。请查看输出列。我想编写一个 python 脚本来准确生成输出列显示的内容。

如您所见,集合“6”有部分“y”,集合 7 也有部分“y”,所以在这种情况下,我想输出为“y, u”,因为它们共享部分“y” .

我希望这是有道理的!

请您参考如下方法:

这更像是网络问题

import networkx as nx 
G=nx.from_pandas_edgelist(df, 'Set', 'Parts') 
l=list(nx.connected_components(G)) 
c1=[[y  for y in x if y in df['Set'].tolist()  ]for x in l] 
c2=[','.join(set([y  for y in x if y in df['Parts'].tolist()]))for x in l] 
from collections import ChainMap 
 
df.Set.map(dict(ChainMap(*map(dict.fromkeys, c1, c2)))) 
Out[167]:  
0     f,a,b,c,d,g,e 
1     f,a,b,c,d,g,e 
2     f,a,b,c,d,g,e 
3     f,a,b,c,d,g,e 
4     f,a,b,c,d,g,e 
5     f,a,b,c,d,g,e 
6     f,a,b,c,d,g,e 
7     f,a,b,c,d,g,e 
8     f,a,b,c,d,g,e 
9     f,a,b,c,d,g,e 
10                z 
11              u,y 
12              u,y 
13              u,y 
Name: Set, dtype: object