Python|熊猫 dataframe.assign()

pandas是Python中非常流行的数据分析库，其提供的DataFrame对象可以方便地进行数据操作和处理。其中，assign()方法是一种非常方便的数据操作方法，可以用于添加、更新或删除DataFrame中的列。

添加列

使用assign()方法可以方便地添加新的列到DataFrame中。通过传递一个或多个新列的名称和相应的计算方式，可以将新列添加到DataFrame中。例如，以下代码将添加一个名为'sum'的新列，该列是'col1'和'col2'列的和：

import pandas as pd

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

df = df.assign(sum=df['col1'] + df['col2'])

print(df)

输出：

   col1  col2  sum
0     1     4    5
1     2     5    7
2     3     6    9

更新列

使用assign()方法也可以方便地更新现有列的值。可以以相同或不同的列名传递计算方式，以更新现有列中的值。例如，以下代码将使用新的计算方式更新'sum'列中的值：

df = df.assign(sum=df['col1'] * df['col2'])

print(df)

输出：

   col1  col2  sum
0     1     4    4
1     2     5   10
2     3     6   18

删除列

最后，assign()方法可以用于方便地删除一个或多个列。可以使用del语句或drop()方法来删除列，但assign()方法提供了一种简单的方法来删除列。只需将要删除的列名称传递给assign()方法即可。例如，以下代码将删除名为'sum'的列：

df = df.assign(sum=None)

print(df)

输出：

   col1  col2   sum
0     1     4  None
1     2     5  None
2     3     6  None

总的来说，assign()方法是一种非常方便的数据操作方法，可以用于添加、更新或删除DataFrame中的列，使得数据处理更加灵活、高效。