1047 votos

python: obtener la lista de los pandas dataframe encabezados de columna

Quiero obtener una lista de los encabezados de columna de una pandas dataframe. El dataframe vendrá de la entrada de usuario, así que no voy a saber cuántas columnas habrá o lo que se llama. Por ejemplo, si me dan un dataframe como este:

my_dataframe =

    y  gdp  cap
0   1    2    5
1   2    3    9
2   8    7    2
3   3    4    7
4   6    7    7
5   4    8    3
6   8    2    8
7   9    9   10
8   6    6    4
9  10   10    7

Me gustaría obtener una lista como esta:

header_list =

[y, gdp, cap]

1690voto

Simeon Visser Puntos 30697

Usted puede obtener los valores como una lista haciendo:

list(my_dataframe.columns.values)

417voto

EdChum Puntos 10205

Hay una construida en el método:

my_dataframe.columns.values.tolist()

.columns devuelve un Index, .columns.values devuelve un array y este tiene una función de ayuda para devolver un list.

EDITAR

Para aquellos que odian escribir este es probablemente el método más corto:

list(df)

25voto

BrenBarn Puntos 63718

Disponible como my_dataframe.columns.

19voto

Sascha Gottfried Puntos 1014
[column for column in my_dataframe]

los pandas docs: la Iteración a través de dataframes volver etiquetas de columna]

1voto

user21988 Puntos 29
n = []
for i in my_dataframe.columns:
    n.append(i)
print n

Iteramos.com

Iteramos es una comunidad de desarrolladores que busca expandir el conocimiento de la programación mas allá del inglés.
Tenemos una gran cantidad de contenido, y también puedes hacer tus propias preguntas o resolver las de los demás.

Powered by:

X