Skip to main content
 首页 » 编程设计

python之Pandas :迭代已经排序的列的唯一值

2025年05月04日19myhome

我已经按排序顺序构建了一个 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