如何使用多个标签访问groupby数据帧中的值？

Question

我试图找到数据帧中已经分组的值

我将付款数据与借钱人借钱的时间、借钱人付款所用的月数进行分组，并将他们支付的金额相加。我的目标是找出人们花了几个月才偿还的清单

例如，当start\u yyyymm为201807时，我如何知道'month\u taken'的列表

payment_sum_monthly = 
payment_data.groupby(['start_yyyymm','month_taken']) 
[['amount']].sum()

如果我使用R并将付款数据放在data.table表格中，我可以找到所用月份的列表

payment_sum_monthly[start_yyyymm == '201807',month_taken]

如何在Python中执行此操作？谢谢

Answer 1

is_date = payment_data['start_yyyymm'] == "201807"

它应该为您提供所有具有'start_yyyymm'is 201807的实体

date_set = payment_data[is_date].copy()
payment_sum_monthly = date_set .groupby('month_taken').aggregate(sum)
payment_sum_monthly 

如果您还需要一个条件，您可以执行以下操作：

condition2 = payment_data['column name'] == condition
payment_data[is_date & condition2]

我希望我把你的问题答对了，这会有帮助