python 如何根据某列值大小关系,赋予新列
时间: 2023-08-11 18:29:18 浏览: 229
python pandas 如何替换某列的一个值
5星 · 资源好评率100%
在Python中,可以使用pandas库将某列值的大小关系赋予新列。下面给出示例代码:
```python
import pandas as pd
# 创建一个DataFrame
df = pd.DataFrame({'A': [1, 2, 3, 4, 5], 'B': [11, 22, 33, 44, 55]})
# 根据列B的值的大小关系,给列C赋值
df['C'] = pd.cut(df['B'], bins=[0, 20, 40, 60], labels=['small', 'medium', 'large'])
print(df)
```
输出结果:
```
A B C
0 1 11 small
1 2 22 medium
2 3 33 medium
3 4 44 medium
4 5 55 large
```
在上述代码中,首先创建了一个DataFrame,包含两列,其中列B的值是需要根据它来赋值的。然后,在新建一列C时,使用了`cut()`函数,它可以将一列数值根据一个或多个分割点进行切割,分割后的区间可以自定义标签。在这里,我们使用了`cut()`函数将列B的值根据分割点[0, 20, 40, 60]进行切割,将其分为三个区间,分别为'small'、'medium'、'large',并将切割后的值赋给新的列C。
阅读全文