python – 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的优先顺序>绿色>黑色,我想消除FRUITS中的所有重复条目

期望的输出应该是：

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

@H_502_18@解决方法:

您可以通过将COLOR设置为有序分类,然后排序并删除重复的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