Erhalten Sie eine Liste von Pandas DataFrame-Spaltenüberschriften

Question

Ich möchte eine Liste der Spaltenüberschriften aus einem Pandas DataFrame erhalten. Das DataFrame wird vom Benutzer eingegeben, daher werde ich nicht wissen, wie viele Spalten es geben wird oder wie sie genannt werden.

Zum Beispiel, wenn ich ein DataFrame wie dieses gegeben habe:

    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

Ich würde eine Liste wie diese erhalten:

['y', 'gdp', 'cap']

Answer 1

%%timeit
final_df.columns.values.tolist()
948 ns ± 19.2 ns pro Schleife (Mittelwert ± Standardabweichung von 7 Durchläufen, 1000000 Schleifen je)

%%timeit
list(final_df.columns)
14.2 µs ± 79.1 ns pro Schleife (Mittelwert ± Standardabweichung von 7 Durchläufen, 100000 Schleifen je)

%%timeit
list(final_df.columns.values)
1.88 µs ± 11.7 ns pro Schleife (Mittelwert ± Standardabweichung von 7 Durchläufen, 1000000 Schleifen je)

%%timeit
final_df.columns.tolist()
12.3 µs ± 27.4 ns pro Schleife (Mittelwert ± Standardabweichung von 7 Durchläufen, 100000 Schleifen je)

%%timeit
list(final_df.head(1).columns)
163 µs ± 20.6 µs pro Schleife (Mittelwert ± Standardabweichung von 7 Durchläufen, 10000 Schleifen je)

Answer 2

Es ist sehr einfach.

Wie zum Beispiel:

list(df.columns)

Answer 3

Die einfachste Option wäre: list(my_dataframe.columns) oder my_dataframe.columns.tolist()

Kein Bedarf für die oben genannten komplexen Sachen :)

Answer 4

Für eine schnelle, ordentliche visuelle Überprüfung, versuchen Sie dies:

für Spalte in df.columns:
    print Spalte

Answer 5

import pandas as pd

# erstelle Test-Datenrahmen
df = pd.DataFrame('x', columns=['A', 'B', 'C'], index=range(2))

list(df.columns)

Gibt zurück

['A', 'B', 'C']