Pandas是一个功能强大的Python库,用于
数据处理
和分析。其中之一的常见操作是在
DataFrame
中添加新的列,并根据某些条件对其进行赋值。在本篇文章中,我们将详细介绍如何使用Pandas新增一列并按条件赋值。
首先,让我们创建一个示例
DataFrame
:
import pandas as pd
data = {'Name': ['Alice', 'Bob', 'Charlie', 'David'],
'Age': [25, 30, 35, 40],
'Gender': ['F', 'M', 'M', 'M']}
df = pd.DataFrame(data)
print(df)
输出结果:
Name Age Gender
0 Alice 25 F
1 Bob 30 M
2 Charlie 35 M
3 David 40 M
我们将向该DataFrame添加一个“Status”列,该列基于以下规则进行赋值:
如果年龄小于30岁,则状态为“Young”。
如果年龄在30到40岁之间,则状态为“Middle”。
如果年龄大于等于40岁,则状态为“Old”。
让我们使用Pandas实现这个需求。
方法1:使用apply方法
我们可以使用Pandas的apply()
方法将自定义函数应用于DataFrame中的每一行。接下来,我们将编写一个名为get_status()
的函数,该函数接受年龄作为输入,并返回相应的状态。然后,我们将调用apply()
方法,以将get_status()
函数应用于DataFrame中的每一行。
def get_status(age):
if age < 30:
return 'Young'
elif age >= 30 and age <= 40:
return 'Middle'
else:
return 'Old'
df['Status'] = df['Age'].apply(get_status)
print(df)
输出结果:
Name Age Gender Status
0 Alice 25 F Young
1 Bob 30 M Middle
2 Charlie 35 M Middle
3 David 40 M Old
因此,我们已经成功地向DataFrame中添加了一个新列,并根据给定的条件对其进行赋值。
方法2:使用numpy.where方法
除了使用apply()
方法外,我们还可以使用NumPy库中的where()
函数将条件应用于整个DataFrame,并为满足该条件的行分配值。
import numpy as np
df['Status'] = np.where(df['Age'] < 30, 'Young',
np.where(df['Age'] <= 40, 'Middle', 'Old'))
print(df)
输出结果与上面相同。这种方法可以更快速地计算,特别是在对大型数据集进行操作时。
本文介绍了两种在Pandas中新增一列并按条件赋值的方法。第一种方法是使用apply()
方法,它非常直观且易于理解。第二种方法使用NumPy的where()
函数,可以更快速地计算,但可能需要一些时间来适应语法。无论您选择哪种方法,都可以根据需要轻松地为Pandas DataFrame添加新列并根据指定条件对其进行赋值。
DataFrame
numpy
pandas