のpython - セット

Question

で最高の加重値を選択するためのエレガントな方法は、私が

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

これらの値はxyz>abc>mnq>qpr は、私は値のセットを持つ列を持つパンダのデータフレームを有するように重み付けされるようなリストを持っています。

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

は今、私はに頼ることなく、この処理を行うためのエレガントな方法があり、私はカスタム関数を適用した後、私は

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

が残っていますようにセットで最高値を選択したいです重みの辞書？

Answer 1

パラメータordered=Trueでpd.Categoricalを使用し、categories=l[::-1]を設定して希望の注文を得ることができます。

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