python - 相当于熊猫中的 fct_lump

Question

Python 中是否有一个函数可以完成 R fct_lump 函数所做的事情（即将所有太小的组分组到一个“其他”组中）？

下面的例子：

library(dplyr)
library(forcats)

> x <- factor(rep(LETTERS[1:9], times = c(40, 10, 5, 27, 1, 1, 1, 1, 1)))

> x
 [1] A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A A B B B B B B B B
[49] B B C C C C C D D D D D D D D D D D D D D D D D D D D D D D D D D D E F G H I
Levels: A B C D E F G H I

> x %>% fct_lump_n(3)
 [1] A     A     A     A     A     A     A     A     A     A     A     A     A     A     A     A    
[17] A     A     A     A     A     A     A     A     A     A     A     A     A     A     A     A    
[33] A     A     A     A     A     A     A     A     B     B     B     B     B     B     B     B    
[49] B     B     Other Other Other Other Other D     D     D     D     D     D     D     D     D    
[65] D     D     D     D     D     D     D     D     D     D     D     D     D     D     D     D    
[81] D     D     Other Other Other Other Other
Levels: A B D Other

标签： pythonrpandasforcats

Answer 1

pip install siuba 
#( in python or anaconda prompth shell)

#use library as:
from siuba.dply.forcats import fct_lump, fct_reorder 

#just like fct_lump of R :

df['Your_column'] = fct_lump(df['Your_column'], n= 10)

df['Your_column'].value_counts() # check your levels

#it reduces the level to 10, lumps all the others as 'Other'