Python Pandas 代码:根据条件更新列内容
可以使用逻辑运算符 & 和 | 来实现对条件的组合。
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 著作权归作者所有。请勿转载和采集!