Python - elegante Art und Weise des höchsten gewichteten Wert im Set zu holen

Question

Ich habe eine Liste, so dass

l = ['xyz','abc','mnq','qpr'] 

Diese Werte so gewichtet werden, dass xyz>abc>mnq>qpr ich Pandas Datenrahmen mit einer Säule, die Sätze von Werten hat.

     COL_NAME  
0   set(['xyz', 'abc'])  
1   set(['xyz']) 
2   set(['mnq','qpr']) 

Nun, ich möchte die höchsten Werte in den Sätzen holen, so dass, nachdem ich die benutzerdefinierte Funktion anwenden ich links bin mit

    COL_NAME  
0   set(['xyz'])  
1   set(['xyz']) 
2   set(['mnq']) 

Gibt es eine elegante Möglichkeit, diesen Prozess zu tun, ohne Rückgriff auf ein Wörterbuch von Gewichten?

Answer 1

können Sie pd.Categorical mit dem Parameter ordered=True verwenden und categories=l[::-1] einstellen, um die Reihenfolge zu erhalten, die Sie möchten.

def max_cat(x): 
    return set([pd.Categorical(x, l[::-1], True).max()]) 

df.COL_NAME.apply(max_cat) 

0 {xyz} 
1 {xyz} 
2 {mnq} 
Name: COL_NAME, dtype: object