python对某一列进行多条件判断,并把判断结果赋予到新增一列中
时间: 2023-06-18 12:08:26 浏览: 172
可以使用 Pandas 库来实现该功能。可以通过 `apply` 方法对某一列进行多条件判断,并将判断结果赋给新增一列。
以下是一个示例代码:
```python
import pandas as pd
# 创建一个示例 DataFrame
df = pd.DataFrame({
'A': [1, 2, 3, 4, 5],
'B': ['a', 'b', 'c', 'd', 'e']
})
# 定义多条件判断函数
def my_func(row):
if row['A'] < 3 and row['B'] == 'a':
return 'True'
elif row['A'] > 4 and row['B'] == 'e':
return 'True'
else:
return 'False'
# 将判断结果赋给新增一列
df['C'] = df.apply(my_func, axis=1)
print(df)
```
输出结果为:
```
A B C
0 1 a True
1 2 b False
2 3 c False
3 4 d False
4 5 e True
```
其中,`my_func` 是自定义的多条件判断函数,`axis=1` 表示按行处理。在函数中,使用 `row['A']` 和 `row['B']` 分别获取每一行中的 A 和 B 列的值,根据条件进行判断并返回 True 或 False。最后,将判断结果赋给新增的 C 列即可。
阅读全文