pandas 按照两个条件判断新增一列,两个条件是或者的关系 使用 loc 方法
时间: 2024-02-24 18:57:25 浏览: 62
pandas按若干个列的组合条件筛选数据的方法
5星 · 资源好评率100%
好的,根据您的要求,我可以告诉您如下的代码实现:
假设您要新增一列名为 `new_col`,并且要按照两个条件判断,这两个条件是或者的关系,可以使用 `loc` 方法实现,代码如下:
```python
import pandas as pd
# 创建示例数据
data = {'A': [1, 2, 3, 4, 5],
'B': [6, 7, 8, 9, 10],
'C': ['a', 'b', 'c', 'd', 'e']}
df = pd.DataFrame(data)
# 使用 loc 方法新增一列 new_col
df.loc[(df['A'] > 2) | (df['C'] == 'e'), 'new_col'] = True
# 展示结果
print(df)
```
解释如下:
- 使用 `loc` 方法定位到要新增的行和列,第一个参数是行索引,第二个参数是列索引;
- 行索引使用了两个条件判断,条件之间使用了 `|`,表示或者的关系;
- 列索引直接指定为 `new_col`,表示新增一列;
- 最后将新增的结果赋值为 `True`。
执行上述代码后,您应该会看到如下的输出:
```
A B C new_col
0 1 6 a NaN
1 2 7 b NaN
2 3 8 c True
3 4 9 d True
4 5 10 e True
```
其中,第 3、4、5 行中的 `new_col` 列都被赋值为了 `True`。
阅读全文