Pandas recherche basée sur la valeur

Question

je la dataframe suivante:

Date best a b c d 
1990 a  5 4 7 2 
1991 c  10 1 2 0 
1992 d  2 1 4 12 
1993 a  5 8 11 6 

Je voudrais faire une trame de données comme suit:

Date best value 
1990 a  5 
1991 c  2 
1992 d  12 
1993 a  5 

Je cherche à trouver une valeur basée sur une autre ligne valeur en utilisant des noms de colonnes. Par exemple, la valeur pour 1990 dans la seconde df devrait rechercher "a" à partir de la première df et la deuxième ligne devrait rechercher "c" (= 2) à partir de la première df.

Des idées?

Answer 1

Vous créez une fonction de recherche et appelez apply sur votre ligne sage dataframe, ce n'est pas très efficace pour les grandes DSF si

In [245]: 

def lookup(x): 
    return x[x.best] 
df['value'] = df.apply(lambda row: lookup(row), axis=1) 
df 
Out[245]: 
    Date best a b c d value 
0 1990 a 5 4 7 2  5 
1 1991 c 10 1 2 0  2 
2 1992 d 2 1 4 12  12 
3 1993 a 5 8 11 6  5 

Answer 2

Il est construit en lookup fonction qui peut gérer ce type de situation (regarde par rangée/colonne). Je ne sais pas comment c'est optimisé, mais peut-être plus rapide que la solution d'application.

In [9]: df['value'] = df.lookup(df.index, df['best']) 

In [10]: df 
Out[10]: 
    Date best a b c d value 
0 1990 a 5 4 7 2  5 
1 1991 c 10 1 2 0  2 
2 1992 d 2 1 4 12  12 
3 1993 a 5 8 11 6  5