web-dev-qa-db-fra.com

Récupérer le nom de la colonne de son index dans Pandas

J'ai un pandas dataframe et un tableau numpy de valeurs de ce dataframe. J'ai l'index d'une colonne spécifique et j'ai déjà l'index de ligne d'une valeur importante. Maintenant je dois obtenir le nom de colonne de cette valeur particulière de mon dataframe.

Après avoir parcouru les documentations, j'ai découvert que je pouvais faire le contraire mais pas ce que je voulais.

pythonpandasnumpyindexing
13
Jatin Bhola

Je pense que vous avez besoin des noms de colonnes d'index par position (le nombre de pythons de 0, donc pour la quatrième colonne il faut 3):

colname = df.columns[pos]

Échantillon:

df = pd.DataFrame({'A':[1,2,3],
                   'B':[4,5,6],
                   'C':[7,8,9],
                   'D':[1,3,5],
                   'E':[5,3,6],
                   'F':[7,4,3]})

print (df)
   A  B  C  D  E  F
0  1  4  7  1  5  7
1  2  5  8  3  3  4
2  3  6  9  5  6  3

pos = 3
colname = df.columns[pos]
print (colname)
D

pos = [3,5]
colname = df.columns[pos]
print (colname)
Index(['D', 'F'], dtype='object')
19
jezrael