熊猫的groupby()和索引值

伊戈尔·里文(Igor Rivin)

我有pandas.DataFrame一个Multiindex,即:

a         val
   dog    1
   cat    2
b         
   fox    3
   rat    4

我想要一个系列,其条目是级别1的索引值列表

所以:

a    [dog, cat]
b    [fox, rat]

下面的方法确实有效,但是速度很慢而且很不优雅:

fff = df.groupby(level=0)['val'].agg(lambda x:[i[1] for i in list(x.index.values)])

所以我希望有更好的方法。

BEN_YO

reset_indexgroupby

df.reset_index(level=1).groupby(level=0)['level_1'].apply(list)


Out[21]: 
a    [dog, cat]
b    [fox, rat]
Name: level_1, dtype: object

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章