pandas 在panda中使用groupby定义聚合函数

dhxwm5r4 于 2023-01-15 发布在其他

关注(0)|答案(2)|浏览(156)

我想使用groupby和agg折叠数据集，但是折叠后，我希望新列只显示分组行的字符串值。例如，初始数据为：

df = pd.DataFrame([["a",1],["a",2],["b",2]], columns=['category','value'])

    category    value
0      a         1
1      a         3
2      b         2

预期输出：

category   value
0     a      grouped
1     b         2

我应该如何修改我的代码（显示"grouped"而不是3）：

df=df.groupby(['category'], as_index=False).agg({'value':'max'})

pandas

来源：https://stackoverflow.com/questions/75090410/defining-an-aggregation-function-with-groupby-in-pandas

2条答案

按热度按时间

wqnecbli1#

可以将lambda与三进制一起使用：

df.groupby("category", as_index=False)
    .agg({"value": lambda x: "grouped" if len(x) > 1 else x})

这将输出：

category    value
0        a  grouped
1        b        2

赞(0）回复(0）举报 2023-01-15

ffx8fchx2#

另一种可能的解决方案：

(df.assign(value = np.where(
    df.duplicated(subset=['category'], keep=False), 'grouped', df['value']))
 .drop_duplicates())

输出：

category    value
0        a  grouped
2        b        2

赞(0）回复(0）举报 2023-01-15

我来回答

pandas 在panda中使用groupby定义聚合函数

2条答案

相关问题

热门标签

最新问答