Numpy 和Python Pandas DataFrame中基于多个条件添加新列

阅读更多：Numpy 教程

介绍

numpy是一个Python中非常强大的数学库，而pandas则是一个非常流行的数据处理的库。在使用pandas DataFrame时，我们经常需要向数据集中添加新的列。这篇文章将通过具体的案例来介绍如何使用numpy和pandas向数据集中添加新列，同时还将讲解如何基于多个条件来进行操作。

创建Pandas DataFrame

首先，我们来创建一个具有一定规模的Pandas DataFrame，这样我们才能更好地演示如何向其中添加新的列以及如何基于多个条件进行操作。

import pandas as pd
import numpy as np

data = {'Name': ['Alex', 'Betty', 'Cherry', 'David', 'Emily', 'Frank', 'Grace', 'Henry', 'Isabella', 'Jack'],
        'Gender': ['M', 'F', 'F', 'M', 'F', 'M', 'F', 'M', 'F', 'M'],
        'Age': [21, 22, 22, 23, 25, 28, 30, 31, 31, 35],
        'Salary': [5000, 6000, 7000, 8000, 9000, 10000, 11000, 12000, 13000, 14000],
        'Education': ['Bachelor', 'Master', 'Bachelor', 'PhD', 'Master', 'Master', 'PhD', 'Bachelor', 'PhD', 'Bachelor']}
df = pd.DataFrame(data)

这里创建了一个包含10行5列的Pandas DataFrame，其中包含了员工的姓名，性别，年龄，工资和教育程度。

向DataFrame添加新的列

在Pandas DataFrame中添加新的列需要使用到assign函数，使用该函数可以快速地对DataFrame进行操作。

我们来看一下具体的案例，假设我们现在要为这个DataFrame加一列，在该列中显示每个人的工资等级（Salary Level），等级分为高、中和低三个级别，具体规则如下：

工资低于8000元的，等级为低
工资在8000元至10000元之间的，等级为中
工资超过10000元的，等级为高

def get_salary_level(salary):
    if salary < 8000:
        return 'Low'
    elif 8000 <= salary <= 10000:
        return 'Medium'
    else:
        return 'High'

df = df.assign(SalaryLevel = df['Salary'].apply(get_salary_level))
print(df)

在这个例子中，我们新建了一个函数get_salary_level，该函数用来计算每个人的薪资等级，最后使用assign函数将其赋值给df中的SalaryLevel列。运行程序后，我们得到了如下的结果：

       Name Gender  Age  Salary Education SalaryLevel
0      Alex      M   21    5000  Bachelor         Low
1     Betty      F   22    6000    Master         Low
2    Cherry      F   22    7000  Bachelor         Low
3     David      M   23    8000       PhD      Medium
4     Emily      F   25    9000    Master      Medium
5     Frank      M   28   10000    Master      Medium
6     Grace      F   30   11000       PhD        High
7     Henry      M   31   12000  Bachelor        High
8  Isabella      F   31   13000       PhD        High
9      Jack      M   35   14000  Bachelor        High

从上面的结果可以看出，我们已经成功地将每个人的工资等级（SalaryLevel）添加到了DataFrame中。

基于多个条件向DataFrame添加新的列

有时我们需要基于多个条件对DataFrame进行操作，例如使用多个条件来计算新列的值。下面我们将用一个案例来阐述这个问题：

假设我们现在要新添加一列AgeLevel，该列显示每个人的年纪等级，等级分为5个级别，具体规则如下：

年龄小于22岁的，等级为最年轻
年龄在22岁至25岁之间的，等级为较年轻
年龄在26岁至30岁之间的，等级为中等年龄
年龄在31岁至35岁之间的，等级为较为成熟
年龄超过35岁的，等级为最成熟

def get_age_level(age):
    if age < 22:
        return '最年轻'
    elif 22 <= age <= 25:
        return '较年轻'
    elif 26 <= age <= 30:
        return '中等年龄'
    elif 31 <= age <= 35:
        return '较为成熟'
    else:
        return '最成熟'

df = df.assign(AgeLevel=np.vectorize(get_age_level)(df['Age']))
print(df)

在这个例子中，我们再次新建了一个函数get_age_level，该函数用来计算每个人的年纪等级，最后使用assign函数将其赋值给df中的AgeLevel列，使用了numpy的vectorize使函数能够向量化计算。运行程序后，我们得到了如下的结果：

       Name Gender  Age  Salary Education SalaryLevel AgeLevel
0      Alex      M   21    5000  Bachelor         Low    最年轻
1     Betty      F   22    6000    Master         Low     较年轻
2    Cherry      F   22    7000  Bachelor         Low     较年轻
3     David      M   23    8000       PhD      Medium     较年轻
4     Emily      F   25    9000    Master      Medium     较年轻
5     Frank      M   28   10000    Master      Medium   中等年龄
6     Grace      F   30   11000       PhD        High   中等年龄
7     Henry      M   31   12000  Bachelor        High    较为成熟
8  Isabella      F   31   13000       PhD        High    较为成熟
9      Jack      M   35   14000  Bachelor        High    最成熟

从上面的结果可以看出，我们已经成功地将每个人的年纪等级（AgeLevel）添加到了DataFrame中。