gpt4 book ai didi

python - 如何给分组添加条件?

转载 作者:太空宇宙 更新时间:2023-11-03 15:43:19 24 4
gpt4 key购买 nike

我有以下数据框:

df = 
GROUP TOTAL_SERVICE_TIME TOTAL_WAIT_TIME IS_EVALUATED IS_NEGATIVE_GRADE
AAA 19 60 0 0
AAA 248 84 1 0
AAA 135 62 1 1
BBB 97 36 1 1
BBB 395 117 0 0

我按如下方式对数据进行分组(按 GROUPTOTAL_WAIT_TIME):

funcs = {
'TOTAL_SERVICE_TIME': {'TOTAL_SERVICE_TIME':'mean'},
'IS_EVALUATED' : {'IS_EVALUATED':'size'},
'IS_NEGATIVE_GRADE' : {'IS_NEGATIVE_GRADE':'size'},
}

fresult = result.groupby(['GROUP','TOTAL_WAIT_TIME']).agg(funcs)
fresult.columns = fresult.columns.droplevel(0)
fresult = fresult.reset_index()
fresult

问题在于 IS_EVALUATEDIS_NEGATIVE_GRADE 计算不正确。我只想计算这些列中 1 的值,而不是所有行。

最佳答案

In [202]: funcs = {
...: 'TOTAL_SERVICE_TIME': 'mean',
...: 'IS_EVALUATED' : 'sum',
...: 'IS_NEGATIVE_GRADE' : 'sum',
...: }
...:
...: fresult = result.groupby(['GROUP','TOTAL_WAIT_TIME'], as_index=False).agg(funcs)
...:

In [203]: fresult
Out[203]:
GROUP TOTAL_WAIT_TIME IS_EVALUATED IS_NEGATIVE_GRADE TOTAL_SERVICE_TIME
0 AAA 60 0 0 19
1 AAA 62 1 1 135
2 AAA 84 1 0 248
3 BBB 36 1 1 97
4 BBB 117 0 0 395

关于python - 如何给分组添加条件?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/41969289/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com