如何在熊猫中按 2 个条件聚合?

时间:2021-06-10 20:33:13

标签: python pandas aggregate

我有一张像下面这样的表格

<头>
日期 训练 条件 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()
)

0 个答案:

没有答案
相关问题