繁体   English   中英

如何在python数据框中获取列的唯一组合并按它们排序?

[英]how to get the unique combination of columns and sort by them in python data frame?

如何在python数据框中获取列的唯一组合并按它们排序? 我知道我可以使用 df.groupby(['col1','col2']).size() 来获得独特的组合。 但是,我也希望按 col2 的升序排列结果顺序,然后按 col1 升序排列。 例如,如果我的数据框是这样的:

  col1  col2
0    A     1
1    B     3
2    C     2
3    D     1
4    A     1
5    F     2

我希望最终输出如下所示:

  col1  col2 
0    A     1 
1    D     1 
2    C     2
3    F     2
4    B     3 

使用groupby通过col2col1 ,但后来需要swaplevel (如果顺序列col1col2是很重要的)与reset_index -在溶液中使用默认排序groupby

df1 = df.groupby(['col2','col1']).size().swaplevel(0,1).reset_index(name='count')
print (df1)
  col1  col2  count
0    A     1      2
1    D     1      1
2    C     2      1
3    F     2      1
4    B     3      1

第二种解决方案首先需要col2col1列中的sort_values ,然后将参数sort=False添加到groupby ,因为默认情况下sort=True

df = df.sort_values(['col2','col1'])
print (df)
  col1  col2
0    A     1
4    A     1
3    D     1
2    C     2
5    F     2
1    B     3

print (df.groupby(['col1','col2'], sort=False).size())
col1  col2
A     1       2
D     1       1
C     2       1
F     2       1
B     3       1
dtype: int64

另一个解决方案是先groupby然后sort_values

df1 = df.groupby(['col1','col2']).size().reset_index(name='count')
print (df1)
  col1  col2  count
0    A     1      2
1    B     3      1
2    C     2      1
3    D     1      1
4    F     2      1

df1 = df1.sort_values(['col2','col1'])
print (df1)
  col1  col2  count
0    A     1      2
3    D     1      1
2    C     2      1
4    F     2      1
1    B     3      1

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM