Pandas 添加新列：两种简单方法 | Python 数据分析

要添加一列数据到 pandas DataFrame 中，可以使用以下方法：

1. 使用索引直接赋值：可以通过索引操作符[]来添加新的列，并赋予相应的值。例如，假设有一个 DataFrame df，可以通过df['new_column'] = new_values来添加名为new_column的新列，并将new_values赋给该列的值。

import pandas as pd

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

# 添加新列
df['C'] = [7, 8, 9]

print(df)

输出:

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

2. 使用 assign() 方法：可以使用 DataFrame 的assign()方法来添加新列。该方法可以在现有 DataFrame 的基础上创建一个新的 DataFrame，并添加指定的列。例如，df = df.assign(new_column=new_values)将创建一个新的 DataFrame df，并添加名为new_column的新列，将new_values赋给该列的值。

import pandas as pd

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

# 添加新列
df = df.assign(C=[7, 8, 9])

print(df)

输出:

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

无论使用哪种方法，都可以添加新列到 pandas DataFrame 中。