创建两行之和的新列，但每两行重复一次

Question

我正在努力在数据框中构建一个附加列，该列是一个时间段内两行的总和。这里附上一张图片：

我想创建一个新列，它是每个 TimePeriodId 的“BillType”列中“IN”和“SA”的生命总和。这样，我将为单个 TimePeriodId 设置一个“总生命数”条目。我已经浏览了很多文档，但无法弄清楚在这种情况下我会怎么做。

代码示例：

sa = pd.read_sql(sa_q1, sql_conn)

#convert TimePeriodId to string values

sa['TimePeriodId'] = sa['TimePeriodId'].astype(str)

sa = sa.loc[(sa['BillType'] =='SA') | (sa['BillType']=='IN')]#.drop(['BillType'], axis = 1)

sa.head(10).to_dict()

#the last line returns the following:

{'TimePeriodId': {1: '201811',

  2: '201811',

  4: '201812',

  5: '201812',

  9: '201901',

  11: '201901',

  13: '201902',

  14: '201902',

  17: '201903',

  18: '201903'},

 'BillType': {1: 'IN',

  2: 'SA',

  4: 'IN',

  5: 'SA',

  9: 'SA',

  11: 'IN',

  13: 'IN',

  14: 'SA',

  17: 'IN',

  18: 'SA'},

 'Lives': {1: 1067,

  2: 288028,

  4: 1058,

  5: 287501,

  9: 293560,

  11: 1068,

  13: 1089,

  14: 278850,

  17: 1076,

  18: 276961}}

任何帮助，将不胜感激！

Answer 1

您可以尝试使用pandas.DataFrame.groupby()方法来计算每个时间段的生命总和。之后，您可以sa使用方法通过计算列丰富数据框pandas.DataFrame.transform()。>>> sa['LivesTotal'] = sa.groupby('TimePeriodId').Lives.transform('sum')