首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >有没有一种替代的、更有效的方法来从pandas数据帧的多索引中拆分列?

有没有一种替代的、更有效的方法来从pandas数据帧的多索引中拆分列?
EN

Stack Overflow用户
提问于 2021-03-01 01:17:10
回答 1查看 88关注 0票数 1

我有一个对象,它是我通过执行groupby(

"A“、"B”

结合熊猫中的.nlargest(3)函数。

即:

代码语言:javascript
复制
df.groupby(["A", "B"])["Column"].nlargest(3).reset_index().unstack()

现在我对每个"A“和"B”都有3个值。我做了一个解栈,它工作了,但我达到了内存容量,它有时会崩溃。

我记得很久以前我找到了这个问题的(内置)解决方案,但再也找不到了。很抱歉,如果这是一个副本,并提前感谢!

EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2021-03-01 05:04:40

据我所知,

在做好一些初步准备后应该会有所帮助

创建数据:

代码语言:javascript
复制
import numpy as np
np.random.seed(2021)
df = pd.DataFrame({'A':np.random.randint(1,3,15), 'B':np.random.randint(1,3,15), 'C':np.random.normal(0,1,15)})
df

看起来像这样

代码语言:javascript
复制
A   B   C
0   1   1   2.044890
1   2   1   1.075268
2   2   1   0.079020
3   1   1   0.493282
4   2   1   -0.791367
5   1   2   -2.130595
6   1   2   0.317206
7   1   2   -1.364617
8   2   2   0.358428
9   1   1   -1.305624
10  2   2   2.020718
11  2   1   -2.686804
12  2   2   0.557872
13  2   1   0.776176
14  1   1   0.202650

然后我们选择3个最大的,

使用

要指定排名,并在排名上进行枢轴操作,请执行以下操作:

代码语言:javascript
复制
df2 = df.groupby(["A", "B"])["C"].nlargest(3).reset_index()
df2['rank'] = df2.groupby(["A", "B"]).cumcount()
pd.pivot_table(df2, values = 'C', index = ['A','B'], columns = 'rank')

这产生了

代码语言:javascript
复制
rank   0           1           2
A   B           
1   1   2.044890    0.493282    0.202650
    2   0.317206    -1.364617   -2.130595
2   1   1.075268    0.776176    0.079020
    2   2.020718    0.557872    0.358428

如果这就是你想要的,如果它能在内存方面工作,请让我知道

票数 2
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/66411797

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档