如何使用 Pandas 在数据框中根据条件创建新列并实现列值累加?

如何使用 pandas 在数据框中根据条件创建新列并实现列值累加?

条件创建新列,实现列值累加条件修改

为了在 pandas 数据框中根据特定条件创建新列,并实现列值的累加,可以采用如下方式:

使用 apply() 方法和 lambda 函数,根据给定条件设置新列的值。
使用 cumsum() 方法计算累加值。
使用 fillna(0) 填充 nan 值。
以下代码实现了上述要求:

import numpy as np
import pandas as pd

# 创建数据框
values = [[5.5, 2.5, 10.0], [2.0, 4.5, 1.0], [2.5, 5.2, 8.0], [4.5, 5.8, 4.8], [4.6, 6.3, 9.6],
           [4.1, 6.4, 9.0], [5.1, 2.3, 11.1]]
df = pd.dataframe(values, columns=['col1', 'col2', 'col3'], index=['a', 'b', 'c', 'd', 'e', 'f', 'g'])

# 根据条件创建新列
df['col4'] = pd.series(np.where((df['col1'] > 3) & (df['col1'] < 5), 1, np.nan), index=df.index).cumsum().fillna(0)

# 显示修改后的数据
print(df)

输出:

   col1  col2  col3  col4
0   5.5   2.5  10.0   0
1   2.0   4.5   1.0   0
2   2.5   5.2   8.0   0
3   4.5   5.8   4.8   1
4   4.6   6.3   9.6   2
5   4.1   6.4   9.0   3
6   5.1   2.3  11.1   0

以上就是如何使用 Pandas 在数据框中根据条件创建新列并实现列值累加?的详细内容,更多请关注硕下网其它相关文章!