代码之家 › 专栏 › 技术社区 › Anakin Skywalker

修复列名并在将数据框按两列分组后重命名

pandas-groupby dataframe pandas python-3.x python

Anakin Skywalker · 技术社区 · 3 年前

我有一个数据帧:

{'ARTICLE_ID': {0: 111, 1: 111, 2: 222, 3: 222, 4: 222}, 'CITEDIN_ARTICLE_ID': {0: 11, 1: 11, 2: 11, 3: 22, 4: 22}, 'enrollment': {0: 10, 1: 10, 2: 10, 3: 10, 4: 10}, 'Trial_year': {0: 2017, 1: 2017, 2: 2017, 3: 2017, 4: 2017}, 'AUTHOR_ID': {0: 'aaa', 1: 'aaa', 2: 'aaa', 3: 'aaa', 4: 'aaa'}, 'AUTHOR_RANK': {0: 1, 1: 2, 2: 3, 3: 4, 4: 5}}

我把它分成两列

df_grouped = df.groupby(['AUTHOR_ID', 'Trial_year']).agg({'ARTICLE_ID': "count", 
                                                      'enrollment': ["count", 'sum']}).reset_index()

因此,我收到了这个数据框,其中列名有两个级别

{('AUTHOR_ID', ''): {0: 'aaa'}, ('Trial_year', ''): {0: 2017}, ('ARTICLE_ID', 'count'): {0: 5}, ('enrollment', 'count'): {0: 5}, ('enrollment', 'sum'): {0: 50}}

我的理想输出——具有一级列名和重命名列名的数据帧

`AUTHOR_ID`, `Trial_year`, `ARTICLE_ID_count`, `enrollment_count`, `enrollment_sum`

1 回复 | 直到 3 年前

enke 3 年前

您可以修改以下列:

df_grouped.columns = [f"{i}_{j}" if j!='' else i for i,j in df_grouped.columns]

或使用 NamedAgg 从一开始:

df_grouped = (df.groupby(['AUTHOR_ID', 'Trial_year'])
              .agg(ARTICLE_ID_count=('ARTICLE_ID', "count"), 
                   enrollment_count=('enrollment','count'),
                   enrollment_sum=('enrollment','sum')).reset_index())

你也可以把字典传给我 groupby.agg 要了解一些简洁的代码:

df_grouped = (df.groupby(['AUTHOR_ID', 'Trial_year'], as_index=False)
              .agg(**{'_'.join(pair): pair for pair in [('ARTICLE_ID', 'count'), 
                                                        ('enrollment','count'), 
                                                        ('enrollment','sum')]}))

输出:

  AUTHOR_ID  Trial_year  ARTICLE_ID_count  enrollment_count  enrollment_sum
0       aaa        2017                 5                 5              50

推荐文章

Google User · Django管理员在`list_display中未显示`creation_date`字段`

4 月前

user29747013 · 如何创建一个新的数据框架,其中包含原始数据框架中列的聚合列?

4 月前

ÎÎÎ½Î· ÎÎ®Î¹Î½Î¿Ï · Python lxml.html语法错误:使用lxml find时XPATH的谓词无效

4 月前

user29715306 · from_users=和chats=电视节目中的差异

4 月前

Redshoe · 当执行numpy.genfromtxt()时,python是否会读取文件的所有行?

4 月前

RASEL MAHMUD · 为什么以及如何在is_even()函数内的IF条件中递归X变量在满足0后递增?[副本]

4 月前

prayner · 更新嵌套字典包含列表中的项

4 月前

Bringo Jr · 我可以在O(n)中解决这个问题吗?

4 月前

Dave · 如何在for循环中修改列表值

5 月前

Shukurullox Komiljonov · 从记录中获得相互和解。使用SQL

5 月前