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