温馨提示×

Python中assign函数怎么使用

小亿
158
2023-10-09 15:58:31
栏目: 编程语言

在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。

0