案例 11: 添加新列 - 使用 assign 方法
知识点讲解
在 Pandas 中，assign 方法是添加新列到 DataFrame 的一种高效方式。这个方法可以使你在不改变原始 DataFrame 的基础上，添加一列或多列。assign 方法特别适用于链式调用，使得数据处理步骤更加清晰和简洁。
示例代码
# 准备数据和示例代码的运行结果，用于案例 11# 示例数据
data = {'Month': ['Jan', 'Feb', 'Mar', 'Apr'],'Sales': [200, 240, 310, 400],'Expenses': [150, 180, 200, 250]
}
df = pd.DataFrame(data)# 使用 assign 方法添加新列
df_with_margin = df.assign(Margin=lambda x: x['Sales'] - x['Expenses'])df_with_margin在这个示例中，我们有一个包含月份、销售额和支出的 DataFrame。我们使用 assign 方法添加了一个新列 Margin，它是通过计算 Sales 和 Expenses 的差值得到的。这里，lambda 函数应用于整个 DataFrame，并对每一行进行计算。
示例代码运行结果
Month Sales Expenses Margin
0 Jan 200 150 50
1 Feb 240 180 60
2 Mar 310 200 110
3 Apr 400 250 150
这个结果显示了添加的新列 Margin，它表示每个月的净收益。assign 方法的优势在于其简洁性和能够保持原始 DataFrame 不变。
