python - Pandas:用于条目的优先级消除重复值

Question

我有一个像这样的 Pandas 日期框:

     FRUITS     COLOURS  
0     apple      red  
1     berry      black         
2     apple      green     
3     grapes     green    
4     apple      black          
5     grapes     red  
6     tomato     black
7     tomato     green  

记住COLORS red > green > black的优先顺序，我想消除FRUITS中的所有重复条目

期望的输出应该是:

     FRUITS    COLOURS  
0     apple     red    
1     berry     black         
2     grapes    red  
3     tomato    green     

Answer 1

您可以通过将 COLOUR 设置为有序分类，然后排序并删除重复的 FRUITS 来设置顺序:

df['COLOURS'] = pd.Categorical(df['COLOURS'], categories=['red','green','black'],ordered=True)

df.sort_values('COLOURS').drop_duplicates('FRUITS').sort_index()

   FRUITS COLOURS
0   apple     red
1   berry   black
5  grapes     red
7  tomato   green