dataframe根据条件生成新一列
时间: 2023-06-01 08:06:59 浏览: 167
对pandas将dataframe中某列按照条件赋值的实例讲解
可以使用 Pandas 库中的 DataFrame 的 loc 方法和条件语句来生成新一列,例如:
```python
import pandas as pd
# 创建一个 DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
# 根据条件生成新一列
df.loc[df['A'] > 1, 'C'] = 'greater than 1'
# 显示 DataFrame
print(df)
```
输出如下:
```
A B C
0 1 4 NaN
1 2 5 greater than 1
2 3 6 greater than 1
```
阅读全文