pandas assign方法

pandas中的assign方法用于将新列添加到DataFrame中。该方法返回一个新的DataFrame，其中包含原始DataFrame中的所有列以及新添加的列。

语法：

DataFrame.assign(**kwargs)

参数：

- **kwargs：新列的名称和值对应的关键字参数。例如，如果要添加一列名为“new_col”，则应使用“new_col=value”的形式指定关键字参数。

示例：

考虑以下示例DataFrame：

import pandas as pd

df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
print(df)

输出：

   A  B
0  1  4
1  2  5
2  3  6

现在，假设我们要添加一个新列，该列是A列和B列之和的两倍。我们可以使用assign方法来实现：

df_new = df.assign(new_col=(df['A'] + df['B']) * 2)
print(df_new)

输出：

   A  B  new_col
0  1  4      10
1  2  5      14
2  3  6      18

可以看到，新列“new_col”已经添加到了DataFrame中，并且其值是A列和B列之和的两倍。