Pandas：按功能分組和聚合

Question

假設我有一個具有以下值的數據框：

    name     start    end     description
0    ag       20       30       None
1    bgb      21       111      'a'
2    cdd      31       101      None
3    bgb      17       19       'Bla'
4    ag       20       22       None

我想按名稱groupby ，然后獲得（ end - start ）值的平均值。

我可以使用mean （ df.groupby(['name'], as_index=False).mean() ）

但是我怎樣才能給均值函數減去兩列（最后 - 首先）？

Answer 1

您可以減去列，然后按列df['name']分組：

df1 = df['end'].sub(df['start']).groupby(df['name']).mean().reset_index(name='diff')
print (df1)
 name  diff
0   ag     6
1  bgb    46
2  cdd    70

新列diff另一個想法：

df1 = (df.assign(diff = df['end'].sub(df['start']))
         .groupby('name', as_index=False)['diff']
         .mean())
print (df1)
  name  diff
0   ag     6
1  bgb    46
2  cdd    70