对于某个DataFrame,取其中某列,判断其中的值小于0,则赋值“不预警”,若大于0小于2,则赋值“C类预警”,若大于2小于4,则赋值“B类预警”,若大于4,则赋值“A类预警”
时间: 2023-11-24 14:07:00 浏览: 56
根据DataFrame某一列的值来选择具体的某一行方法
5星 · 资源好评率100%
可以使用 Pandas 库来实现这个功能,具体代码如下:
```python
import pandas as pd
# 假设DataFrame名为df,需要判断的列名为col_name
col_name = 'column_name'
# 将小于0的值赋值为“不预警”
df.loc[df[col_name] < 0, col_name] = '不预警'
# 将大于等于0且小于2的值赋值为“C类预警”
df.loc[(df[col_name] >= 0) & (df[col_name] < 2), col_name] = 'C类预警'
# 将大于等于2且小于4的值赋值为“B类预警”
df.loc[(df[col_name] >= 2) & (df[col_name] < 4), col_name] = 'B类预警'
# 将大于等于4的值赋值为“A类预警”
df.loc[df[col_name] >= 4, col_name] = 'A类预警'
```
其中,`df.loc` 是 Pandas 中的一种索引方式,用于定位符合条件的行和列。上面的代码按照题目要求分别对不同的值进行赋值,最终得到的结果会保存在原 DataFrame 中。
阅读全文