Pandas 追加列：使用 assign 方法轻松添加新数据

使用 Pandas 的 assign 方法，你可以轻松地将一列数据追加到 DataFrame 中。

例如，如果我们有一个名为 df 的 DataFrame 对象，我们可以使用以下代码将一列名为 'new_column' 的数据追加到 DataFrame 中：

import pandas as pd

# 创建一个 DataFrame 对象
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})

# 追加一列数据
df = df.assign(new_column=[7, 8, 9])

# 打印结果
print(df)

输出结果：

   A  B  new_column
0  1  4           7
1  2  5           8
2  3  6           9

在上面的代码中，我们使用 assign 方法来创建一个新的 DataFrame 对象，其中包含原始 DataFrame 的所有列和新的 'new_column' 列。我们将新列的值设置为 [7, 8, 9]。最后，我们将新的 DataFrame 对象分配给原始 DataFrame 对象，以覆盖原始 DataFrame，从而实现追加列的目的。

assign 方法是 Pandas 中一个强大的工具，可以帮助你轻松地修改和扩展你的 DataFrame。