python - 如何将 Pandas 中的数据帧相加超过 5 个

Question

在我的天数列中，如果我的数字超过 5，我想对该列求和
输入:

Days  files 
1      10
3      20
4      30
5      40
6      50

所需输出:

Days  files 
1      10
3      20
4      30
5+     90

Answer 1

你可以试试series.clip剪辑系列中的上限，然后进行分组和求和:

out = df['files'].groupby(df['Days'].clip(upper=5)).sum().reset_index()
print(out)

  Days  files
0    1     10
1    3     20
2    4     30
3    5     90

如果您真的想将 5 以上的任何内容更改为 str 类型，您可以 jst 替换 5使用上述逻辑:

out = df['files'].groupby(df['Days'].clip(upper=5).replace(5,'5+')).sum().reset_index()
print(out)

  Days  files
0    1     10
1    3     20
2    4     30
3   5+     90