dataframe增加列并条件赋值
时间: 2023-09-21 10:01:07 浏览: 433
对pandas将dataframe中某列按照条件赋值的实例讲解
要在DataFrame中增加一列并进行条件赋值,可以使用.loc方法。首先创建一个新列,并使用.loc方法将其添加到DataFrame中。然后,可以使用条件语句对新列进行赋值。
具体步骤如下:
1. 使用pandas库导入DataFrame对象。
2. 创建一个空的DataFrame,并添加所需的列。
3. 使用.loc方法将新列添加到DataFrame中。
4. 使用条件语句对新列进行赋值。
示例代码如下所示:
```python
import pandas as pd
# 创建一个空的DataFrame
df = pd.DataFrame()
# 添加所需的列
df['A'] = [1, 2, 3, 4, 5]
df['B'] = [6, 7, 8, 9, 10]
# 使用.loc方法添加新列
df.loc[:, 'C'] = None
# 使用条件语句对新列进行赋值
df.loc[df['A'] > 3, 'C'] = '条件满足'
df.loc[df['A'] <= 3, 'C'] = '条件不满足'
print(df)
```
输出结果如下:
```
A B C
0 1 6 条件不满足
1 2 7 条件不满足
2 3 8 条件不满足
3 4 9 条件满足
4 5 10 条件满足
```
通过以上步骤,我们可以在DataFrame中增加新列并使用条件语句对其进行赋值。
阅读全文