博客

Python中的assign()函数属于pandas库中DataFrame类型的函数，用于对数据进行赋值操作，可以对单个或多个DataFrame的元素同时进行赋值。

具体使用方法如下：

1. 单个元素赋值：

df.assign(column_name=value)

该语句会将DataFrame中某个指定的column的所有对应值都赋为value。可以简单理解为更改该column的全部元素的值。

例：

import pandas as pd

data = {'name': ['Lily', 'Lucy', 'John'], 
            'age': [23, 24, 25]}

df = pd.DataFrame(data)

new_df = df.assign(age=27)   #更改age这个字段中的所有数据为27

print(new_df)

输出结果为：

   name  age
0  Lily   27
1  Lucy   27
2  John   27

1. 多个元素赋值：

df.assign(column1=value1, column2=value2)

该语句会将DataFrame中多个指定的column的所有对应值都进行赋值操作。可以将多个column以逗号分隔的形式列出来，每一列都使用赋值符号=进行操作。

例：

import pandas as pd

data = {'name': ['Lily', 'Lucy', 'John'], 
            'age': [23, 24, 25],
            'sex': ['M', 'F', 'M']}

df = pd.DataFrame(data)

new_df = df.assign(age=27, sex='F')   #同步将age字段所有值更改为27，sex字段所有值更改为'F'

print(new_df)

输出结果为：

   name  age sex
0  Lily   27   F
1  Lucy   27   F
2  John  27   F

以上就是Python中assign()函数的使用方法，可以简化代码编写，提高运行效率，利用该函数可以根据需要进行针对性的更改数据操作。