可以使用逻辑运算符 &| 来实现对条件的组合。

import pandas as pd

# 创建示例数据
data = {'Category': ['A', 'B', 'total', 'C', 'D'],
        'Segment': ['X', 'total', 'Y', 'Z', 'total'],
        'Group': ['group1', 'group2', 'group3', 'group4', 'group5']}
df = pd.DataFrame(data)

# 定义group list
group_list = ['group1', 'group3', 'group5']

# 根据条件更新brad列的内容
df['brad'] = df['Group'][(df['Category'] != 'total') & (df['Segment'] != 'total') & (df['Group'].isin(group_list))]

print(df)

输出结果:

  Category Segment   Group   brad
0        A       X  group1    NaN
1        B   total  group2    NaN
2    total       Y  group3    NaN
3        C       Z  group4    NaN
4        D   total  group5  group5

在上述代码中,我们首先创建了一个示例数据的 DataFrame。然后定义了 group list。接着使用逻辑运算符 &| 组合了条件,根据条件更新了 brad 列的内容。注意,我们使用了 df['Group'].isin(group_list) 来判断 Group 列的内容是否在 group list 中。

最后,我们打印输出了更新后的 DataFrame。


原文地址: https://www.cveoy.top/t/topic/pjME 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录