python中将dataframe中的某列与某个值比较,若相等则赋值1,否则赋值0
时间: 2024-02-17 15:03:50 浏览: 110
对pandas将dataframe中某列按照条件赋值的实例讲解
可以使用 `apply()` 方法实现。假设要将 DataFrame 中的列 `col1` 与值 `value` 比较,若相等则赋值1,否则赋值0,代码如下:
```python
import pandas as pd
df = pd.DataFrame({'col1': ['value', 'value', 'other', 'value']})
def compare_value(x):
if x == 'value':
return 1
else:
return 0
df['new_col'] = df['col1'].apply(compare_value)
print(df)
```
输出结果为:
```
col1 new_col
0 value 1
1 value 1
2 other 0
3 value 1
```
这里定义了一个函数 `compare_value()`,用于将 `col1` 中的值与 `value` 进行比较并返回 1 或 0。然后使用 `apply()` 方法将该函数应用到 `col1` 列中的每一个元素上,最终将结果赋值给新的一列 `new_col`。
阅读全文