我已经按排序顺序构建了一个 pandas 数据框,并希望遍历具有特定列的相同值的组。在我看来,groupby 功能对此很有用,但据我所知,执行 groupby 并不能保证 key 的顺序。如何按排序顺序提取未排序的列值。
这是一个示例数据框:
Foo,1
Foo,2
Bar,2
Bar,1
我想要一个列表 ["Foo","Bar"],其中的顺序由原始数据框的顺序保证。然后我可以使用这个列表来提取适当的行。在我的案例中,排序实际上是由数据框中也给出的列定义的(不包括在上面的示例中),因此如果无法直接提取信息,则可以接受重新排序的解决方案。
请您参考如下方法:
如评论中所述,您可以在保留顺序的列上使用 unique(与 numpy 的 unique 不同,它不排序):
In [11]: df
Out[11]:
0 1
0 Foo 1
1 Foo 2
2 Bar 2
3 Bar 1
In [12]: df[0].unique()
Out[12]: array(['Foo', 'Bar'], dtype=object)
然后你可以使用groupby的get_group访问相关的行:
In [13]: g = df.groupby([0])
In [14]: g.get_group('Foo')
Out[14]:
0 1
0 Foo 1
1 Foo 2
