Pandas中增加一个新列的方法

2023-02-24 00:00:00 pandas 增加 方法

在 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 已经添加到了数据框中。

相关文章