python - Pandas GroupBy 列出一列列表中的值并查找它们的平均值

Question

我有一个大型 pandas 数据框(1M 行)，格式如下:

data = {
    'names': {0: ['Lily', 'Kerry', 'Mona'], 1: ['Kerry', 'Mona'], 2: ['Mona']},
    'sentiment': {0: 10, 1: 2, 2: 0}
}
df = pd.DataFrame(data)
df

                 names  sentiment
0  [Lily, Kerry, Mona]         10
1        [Kerry, Mona]          2
2               [Mona]          0

我想计算名称列中每个唯一名称的平均情绪，结果如下:

   names  sentiment
0   Lily         10
1  Kerry          6
2   Mona          4

唯一名称的数量非常长，因此效率很重要

Answer 1

这需要首先对“name”列进行爆炸，然后是标准的GroupBy.mean():

df.explode('names').groupby('names', as_index=False, sort=False).mean()

   names  sentiment
0   Lily         10
1  Kerry          6
2   Mona          4