python df 如何增加一列

原创

mob64ca12f8da8d 2023-12-19 07:02:56 ©著作权

文章标签 解决方案 python Python 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12f8da8d的原创作品，请联系作者获取转载授权，否则将追究法律责任

使用 Python DataFrame 增加一列的方案

问题描述

假设我们有一个数据集，包含学生的姓名、年龄和成绩。我们想要在数据集中增加一列，表示学生的等级。等级的划分规则如下：

成绩大于等于90分，等级为"A"
成绩大于等于80分，等级为"B"
成绩大于等于70分，等级为"C"
成绩大于等于60分，等级为"D"
其他情况，等级为"E"

我们想要使用 Python 的 DataFrame 对象来解决这个问题。

解决方案

我们可以使用 Pandas 库来操作 DataFrame 对象，并通过一些条件判断语句来计算学生的等级。下面是具体的解决方案：

首先，我们需要导入 Pandas 库和创建一个包含学生数据的 DataFrame 对象。

import pandas as pd

# 创建 DataFrame 对象
data = {'姓名': ['张三', '李四', '王五', '赵六'],
        '年龄': [18, 19, 20, 21],
        '成绩': [85, 92, 78, 66]}

df = pd.DataFrame(data)

接下来，我们可以使用 Pandas 的 apply 方法来将一个函数应用到 DataFrame 的每一行上，并返回一个新的列。

# 定义一个函数，用于计算学生的等级
def calculate_grade(score):
    if score >= 90:
        return "A"
    elif score >= 80:
        return "B"
    elif score >= 70:
        return "C"
    elif score >= 60:
        return "D"
    else:
        return "E"

# 将函数应用到 DataFrame 的 '成绩' 列上，并返回一个新的列
df['等级'] = df['成绩'].apply(calculate_grade)

我们可以使用 head 方法来查看 DataFrame 的前几行，以验证新的一列是否添加成功。

# 查看 DataFrame 的前几行
print(df.head())

输出结果如下：

   姓名  年龄  成绩 等级
0  张三  18  85  B
1  李四  19  92  A
2  王五  20  78  C
3  赵六  21  66  D

从输出结果可以看出，新的一列已成功添加到 DataFrame 对象中。

类图

下面是使用 Mermaid 语法表示的类图，展示了上述解决方案中涉及的类和方法。

classDiagram
    class DataFrame {
        - data: dict
        + __init__(data: dict)
        + apply(func) -> Series
        + head() -> DataFrame
    }
    class Series {
        - data: list
        + __init__(data: list)
    }
    class pd {
        + DataFrame
        + Series
    }
    DataFrame --> pd
    Series --> pd