如何将计算应用于数据框中的groupby行，并将结果附加到数据框的底部？

Question

我试图将一个简单计算的结果附加到一个数据帧中。我想按年份分组，然后计算收入减去税收，得到一组新的记录，称为净收入。你知道吗

import pandas as pd data = {'year': [2010, 2011, 2012, 2011, 2012, 2010], 'item': ['Revenue', 'Revenue', 'Revenue', 'Tax', 'Tax', 'Tax'], 'value': [11, 8, 10, 3, 2, 3]} dfRev = pd.DataFrame(data, columns=['year', 'item', 'value']) print dfRev

Answer 1

你可以试试旋转：

#reshape so now you have a col for Tax and a col for Revenue
pivot = dfRev.pivot('year', 'item', 'value')
#perform the calculation
pivot.loc[:, 'Net Revenue'] = pivot.Revenue - pivot.Tax
#and then bring it back to the original shape
dfRev = pivot.stack().reset_index().sort_values(by='item')

print dfRev
   year         item   0
2  2010  Net Revenue   8
5  2011  Net Revenue   5
8  2012  Net Revenue   8
0  2010      Revenue  11
3  2011      Revenue   8
6  2012      Revenue  10
1  2010          Tax   3
4  2011          Tax   3
7  2012          Tax   2