python - 从 groupby 索引中的拆分字符串聚合值

Question

我有一个如下所示的数据框:

index  name     value
1      Foo      9
2      Bar      11
3      Foo      2
4      Bar      4
5      Foo,Bar  3

如果我执行 df.groupby('name').count() 我得到:

         value
name
Foo      2
Bar      2
Foo,Bar  1

我想拆分 Foo,Bar 并将结果元组中存在的索引中的每个值加 1，这样分组的数据框如下所示:

         value
name
Foo      3
Bar      3

我可以想办法用 for 循环来做到这一点，但我想知道是否有一些内置的 pandas 方法可以解决这类问题。

Answer 1

对于 pandas 0.25+ 使用 Series.str.split与 Series.explode对于 flatten Series，然后对于计数 Series.value_counts与 Series.to_frame :

df1 = df['name'].str.split(',').explode().value_counts().to_frame('value')
print (df1)
     value
Foo      3
Bar      3

对于另一个版本，对 DataFrame 和 DataFrame.stack 使用 split 和 expand=True对于 flatten Series:

df1 = df['name'].str.split(',', expand=True).stack().value_counts().to_frame('value')
print (df1)
     value
Foo      3
Bar      3