首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往

pandas dataframe创建一个新列,该列的值基于另一列上的groupby sum

在pandas中,可以使用groupbysum函数来创建一个新列,该列的值基于另一列上的分组求和。

首先,我们需要导入pandas库:

代码语言:python
复制
import pandas as pd

然后,我们可以创建一个包含数据的DataFrame对象。假设我们有一个名为data的DataFrame,其中包含两列column1column2

代码语言:python
复制
data = pd.DataFrame({'column1': [1, 2, 3, 4, 5],
                     'column2': [10, 20, 30, 40, 50]})

接下来,我们可以使用groupby函数按照column1列进行分组,并使用sum函数对column2列进行求和。将结果赋值给一个新列new_column

代码语言:python
复制
data['new_column'] = data.groupby('column1')['column2'].transform('sum')

这样,我们就创建了一个新列new_column,其值基于column1列上的分组求和。

关于pandas DataFrame的更多信息,你可以参考腾讯云的产品介绍链接:腾讯云·Pandas DataFrame

页面内容是否对你有帮助?
有帮助
没帮助

相关·内容

没有搜到相关的视频

领券