在 pandas 中,可以使用 DataFrame 对象的 assign 方法或直接对列进行赋值来添加新的列。
1、使用 assign 方法
使用 assign 方法可以将新的列添加到数据框中,并返回一个新的数据框,例如:
import pandas as pd
# 创建一个数据框
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
print(df)
# 添加新的列
df_new = df.assign(C=[7, 8, 9])
print(df_new)
输出:
A B
0 1 4
1 2 5
2 3 6
A B C
0 1 4 7
1 2 5 8
2 3 6 9
可以看到,新的列 C 已经添加到了数据框中。
直接对列进行赋值
也可以直接对列进行赋值,例如:
import pandas as pd
# 创建一个数据框
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
print(df)
# 添加新的列
df['C'] = [7, 8, 9]
print(df)
输出:
A B
0 1 4
1 2 5
2 3 6
A B C
0 1 4 7
1 2 5 8
2 3 6 9
可以看到,新的列 C 已经添加到了数据框中。