如何通过键访问groupby对象中相应的groupby Dataframe ?
使用以下groupby:
rand = np.random.RandomState(1)
df = pd.DataFrame({'A': ['foo', 'bar'] * 3,
'B': rand.randn(6),
'C': rand.randint(0, 20, 6)})
gb = df.groupby(['A'])
我可以迭代它来获取键和组:
In [11]: for k, gp in gb:
print 'key=' + str(k)
print gp
key=bar
A B C
1 bar -0.611756 18
3 bar -1.072969 10
5 bar -2.301539 18
key=foo
A B C
0 foo 1.624345 5
2 foo -0.528172 11
4 foo 0.865408 14
我希望能够通过组的密钥访问组:
In [12]: gb['foo']
Out[12]:
A B C
0 foo 1.624345 5
2 foo -0.528172 11
4 foo 0.865408 14
但是当我尝试用gb[('foo',)]
做这个的时候,我得到了一个奇怪的pandas.core.groupby.DataFrameGroupBy
对象,它似乎没有任何对应于我想要的DataFrame的方法。
我能想到的最好的办法是:
In [13]: def gb_df_key(gb, key, orig_df):
ix = gb.indices[key]
return orig_df.ix[ix]
gb_df_key(gb, 'foo', df)
Out[13]:
A B C
0 foo 1.624345 5
2 foo -0.528172 11
4 foo 0.865408 14
但考虑到Pandas通常在这些事情上是多么的友好,这有点讨厌。
做这件事的内在方法是什么?
6条答案
按热度按时间u5rb5r591#
可以使用
get_group
方法:dict(iter(gb))
创建朴素字典更节省内存。这是因为它使用了groupby对象中已有的数据结构。*您可以使用groupby切片选择不同的列:
cwxwcias2#
Wes McKinney(pandas的作者)在Python for Data Analysis中提供了以下配方:
它返回一个字典,其键是您的组标签,其值是DataFrames,即
将产生您正在寻找的内容:
1wnzp6jl3#
而不是
我更喜欢使用
gb.groups
因为这样你也可以选择多列。例如:
6ljaweal4#
如果要查找选择性groupby对象,请执行以下操作:gb_groups.keys(),并将所需密钥输入到以下密钥列表中。
n3h0vuf25#
我正在寻找一种方法来抽样的几个成员的GroupBy obj -不得不解决张贴的问题,以完成这一点。
根据
some_key
列创建groupby对象选择N个 Dataframe 并获取其索引
抓取分组
可选-将其全部转换回单个 Dataframe 对象
zvokhttg6#
相当于: