我有一张像下面这样的表格
日期 | 训练 | 条件 1 | 条件2 |
---|---|---|---|
第一天 | 1111 | 真实 | 0 |
第2天 | 2222 | 假 | 2 |
.....
我想将火车与日期分组,
日期 | 列车总数 | 条件 1 的训练为真 | 训练条件 2 > 0 | 训练条件1为真,条件2>0 |
---|---|---|---|---|
第一天 | 1 | 1 | 0 | 0 |
第2天 | 1 | 0 | 1 | 0 |
我写了一个像下面这样的函数,发现它似乎不能支持 unhashable 类型:'list'?
df1 = (
df.groupby('date')
.agg(**{
'train total count': ('train', 'count'),
'train condition1 count': ('conditon1', lambda x: sum(x.gt(0))),
'train condition1 &2 count': (['condition1', 'condition2'], lambda x: sum(x[0] + x[1] > 0))
}).reset_index()
)