如何根据pandas中的多个条件生成一列?

蟒蛇菜鸟

我想要给定问题的python代码。 这是 CSV 格式的数据集,如下所示:

ID   SCORE
 1     3
 1     5
 1     8
 1     1
 2     6
 2     0
 2     3
 1     6
 1     4

从这个数据集中,我想创建一个名为“ OUTPUT的列条件是当ID=1 时应添加SCORE的每一行,直到ID=2并且应将总和插入到OUTPUT列中。ID=2时,将应用类似的逻辑。这是我想要的输出:

ID   SCORE  OUTPUT
 1     3      17 
 1     5      17 
 1     8      17 
 1     1      17 
 2     6      9 
 2     0      9 
 2     3      9 
 1     6      10
 1     4      10
奥马尔.H

创建自定义分组然后使用.transformsum

df['OUTPUT'] = df.groupby(df['ID'].ne(df['ID'].shift())\
                 .cumsum())['SCORE'].transform('sum')


print(df)

   ID  SCORE  OUTPUT
0   1      3      17
1   1      5      17
2   1      8      17
3   1      1      17
4   2      6       9
5   2      0       9
6   2      3       9
7   1      6      10
8   1      4      10

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章

如何根据Pandas中另一列的条件比较同一列中的日期?

如何根据熊猫另一列中的条件生成具有值的新列

如何根据另一列中的多个条件创建新列

根据条件用一列中的多个元素替换

如何根据条件在熊猫中创建另一列?

根据另一列的条件更新Pandas DataFrame中的一列

如何根据另一列中的条件转置一列?

如何根据另一列的条件/分组删除一列中的连续重复行?

如何根据条件将一列中的值设置为另一列

如何根据 Pandas 数据框中另一列的条件将值填充到新列中?

如何在SQL中查询一列的多个条件?

根据另一列中的条件填充多个数据框列

根据另一列的条件汇总一列的多个连续行中的字符串数据

根据Postgres中另一列的条件生成整数系列

SQL查询根据条件选择一列的多个实例

根据多个条件合并行,然后对一列求和

根据一列中的多个条件删除行(使用Python)

pandas 根据另一列中的条件从 1 列中提取公共子串

Pandas 根据另一列的条件有选择地覆盖列中的值

根据另一列中的条件在 Pandas 数据框中设置值

Pandas - 根据后一列中是否存在值过滤一列中具有相同值和另一列中多个值的行

如何根据另一列中的条件更改数据框的某一列中的值?

如何根据另一列中的值进行条件替换列中的值?

如何根据数据框另一列中的条件在列中查找最小值?

我如何获得条件中R中多列的中位数(根据另一列)

如何根据另一列中的条件汇总列中的值?

如何根据另一列中满足的条件在熊猫数据框中添加新列?

Python Pandas:根据另一列更改条件

根据Python中的多个条件,将一列从多个数据框合并到另一数据框