如何在Pandas中创建条件文本列？

在Pandas中创建条件文本列可以通过使用np.where()函数或DataFrame.apply()方法来实现。

首先，导入必要的库：

import pandas as pd
import numpy as np

然后，创建一个示例DataFrame：

data = {'Name': ['Alice', 'Bob', 'Charlie', 'David'],
        'Age': [25, 30, 35, 40],
        'Score': [80, 90, 85, 95]}
df = pd.DataFrame(data)

接下来，使用np.where()函数创建条件文本列：

df['Grade'] = np.where(df['Score'] >= 90, 'A', 'B')

上述代码中，如果分数大于等于90，则将Grade列的值设置为'A'，否则设置为'B'。

最后，打印DataFrame以查看结果：

print(df)

输出结果如下：

      Name  Age  Score Grade
0    Alice   25     80     B
1      Bob   30     90     A
2  Charlie   35     85     B
3    David   40     95     A

使用DataFrame.apply()方法： DataFrame.apply()方法可以对DataFrame的每一行或每一列应用自定义函数，从而创建条件文本列。以下是创建条件文本列的步骤：

首先，创建一个自定义函数，根据条件返回相应的文本：

def get_grade(score):
    if score >= 90:
        return 'A'
    else:
        return 'B'

然后，使用DataFrame.apply()方法将该函数应用于DataFrame的某一列：

df['Grade'] = df['Score'].apply(get_grade)

上述代码中，将自定义函数get_grade()应用于Score列。

最后，打印DataFrame以查看结果：

print(df)

输出结果与上述方法相同：

      Name  Age  Score Grade
0    Alice   25     80     B
1      Bob   30     90     A
2  Charlie   35     85     B
3    David   40     95     A

以上是在Pandas中创建条件文本列的两种常见方法。根据实际需求选择适合的方法即可。

腾讯云相关产品和产品介绍链接地址：

扫码

添加站长进交流群

领取专属 10元无门槛券

手把手带您无忧上云