将dataframe中某一列的值大于0.5的话并在新的列种做标记
时间: 2024-03-15 21:47:07 浏览: 77
好的,您可以使用 Pandas 库中的 apply 函数和 lambda 表达式来实现这个功能。代码如下:
```python
import pandas as pd
# 创建示例数据
data = {'A': [0.3, 0.6, 0.8, 0.1], 'B': [0.4, 0.2, 0.7, 0.9]}
df = pd.DataFrame(data)
# 使用 apply 函数和 lambda 表达式给大于 0.5 的值打标记
df['C'] = df['A'].apply(lambda x: 'Y' if x > 0.5 else 'N')
print(df)
```
运行结果如下:
```
A B C
0 0.3 0.4 N
1 0.6 0.2 Y
2 0.8 0.7 Y
3 0.1 0.9 N
```
以上代码中,我们首先使用 Pandas 的 DataFrame 创建了一个示例数据,包含两列 A 和 B。然后,我们使用 apply 函数和 lambda 表达式,对 A 列中大于 0.5 的值打标记,标记为 Y,否则为 N,并将结果存储到新的列 C 中。最后,我们输出整个 DataFrame,查看结果。
阅读全文