Python中assign函数怎么使用

在Python中，assign函数不是Python内置函数，而是pandas库中的一个方法。assign方法用于在DataFrame中添加新的列或修改已有列的值。

assign方法的用法如下：

df.assign(**kwargs) 

其中，df是一个DataFrame对象，kwargs是一个或多个关键字参数，每个关键字参数是一个字典，字典的键是要创建或修改的列名，字典的值是要为该列设置的值。

示例：

import pandas as pd # 创建一个DataFrame df = pd.DataFrame({'A': [1, 2, 3]}) # 使用assign方法添加新的列 df = df.assign(B=[4, 5, 6]) print(df) 

输出结果：

A  B 0  1  4 1  2  5 2  3  6 

在上述示例中，使用assign方法向DataFrame中添加了一个名为B的新列，并设置了该列的值。

除了可以添加新列，assign方法还可以用于修改已有列的值。例如：

import pandas as pd # 创建一个DataFrame df = pd.DataFrame({'A': [1, 2, 3]}) # 使用assign方法修改已有列的值 df = df.assign(A=df['A'] * 2) print(df) 

输出结果：

A 0  2 1  4 2  6 

在上述示例中，使用assign方法修改了已有列A的值，将其乘以2。