pandas 比较两个字段if 判断 返回一个新的值
时间: 2023-12-19 18:07:36 浏览: 76
两字符串比较返回重复个数
你可以使用 `numpy.where` 函数来比较两个字段并返回一个新的值。具体步骤如下:
假设你要比较的两个字段分别是 df['col1'] 和 df['col2'],如果 df['col1'] 大于 df['col2'],则返回字符串 'Yes',否则返回字符串 'No',你可以使用以下代码:
```python
import pandas as pd
import numpy as np
df = pd.DataFrame({'col1': [1, 2, 3, 4], 'col2': [2, 3, 1, 5]})
df['new_col'] = np.where(df['col1'] > df['col2'], 'Yes', 'No')
print(df)
```
输出结果如下:
```
col1 col2 new_col
0 1 2 No
1 2 3 No
2 3 1 Yes
3 4 5 No
```
在上面的代码中,我们使用了 `np.where` 函数来比较 df['col1'] 和 df['col2'],如果 df['col1'] 大于 df['col2'],则返回字符串 'Yes',否则返回字符串 'No'。最后,我们将新的列添加到原始数据框中,并将其命名为 'new_col'。
阅读全文