Python 数据帧 : Merge values of columns according to a specific condition

Question

您好，我有一个数据框问题。假设我有一个格式如下的数据框

label    value 
1        a
1        b 
2  
2 
1        c
1        d

所以现在我有标签 1 的两个连续部分。我想要这样的输出:

output: [ab,cd] 

不同区域的标签1的哪些相关值被合并在一起。谢谢。

Answer 1

您可以使用itertools.groupby ，仅对相似的相邻项目进行分组:

from itertools import groupby
from operator import itemgetter

zipper = zip(df['label'], df['value'])
grouper = groupby(list(zipper), key=itemgetter(0))
res = [''.join(map(itemgetter(1), j)) for i, j in grouper if i == 1]

['ab', 'cd']