Wie extrahiere ich Spaltenüberschriften aus Pandas DataFrame mit Benutzereingaben?

Spaltenüberschriften aus Pandas DataFrame extrahieren

Das Abrufen einer Liste von Spaltenüberschriften aus einem Pandas DataFrame ist ein häufiger Vorgang für die Datenanalyse. In diesem Artikel zeigen wir, wie dies erreicht wird, wenn der DataFrame aus Benutzereingaben generiert wird, um die Kompatibilität mit einer unbekannten Anzahl oder Namen von Spalten sicherzustellen.

DataFrame Column Header Extraction

Um die Liste der Spaltenüberschriften aus einem DataFrame abzurufen, können Sie Folgendes verwenden:

• columns.values: Dieses Attribut gibt ein Array von Spaltenbezeichnungen zurück, das kann mit list(my_dataframe.columns.values) in eine Liste konvertiert werden.
• Direktes Casting: Alternativ können Sie den DataFrame einfach mit list(my_dataframe) in eine Liste umwandeln. Dies führt zu eine Liste von Spaltenüberschriften, gefolgt von den DataFrame-Werten.

Beispiel

Betrachten Sie den folgenden DataFrame:

<code class="python">import pandas as pd

data = {
    'y': [1, 2, 8, 3, 6, 4, 8, 9, 6, 10],
    'gdp': [2, 3, 7, 4, 7, 8, 2, 9, 6, 10],
    'cap': [5, 9, 2, 7, 7, 3, 8, 10, 4, 7]
}

df = pd.DataFrame(data)</code>

Erhalten von Spaltenüberschriften

Verwendung der columns.values-Methode:

<code class="python">headers = list(df.columns.values)
print(headers)

# Output: ['y', 'gdp', 'cap']</code>

Verwendung von Direktumwandlung:

<code class="python">headers = list(df)
print(headers)

# Output: ['y', 'gdp', 'cap']</code>

Beide Ansätze stellen eine Liste von Spaltenüberschriften bereit : ['y', 'BIP', 'cap'].

Das obige ist der detaillierte Inhalt vonWie extrahiere ich Spaltenüberschriften aus Pandas DataFrame mit Benutzereingaben?. Für weitere Informationen folgen Sie bitte anderen verwandten Artikeln auf der PHP chinesischen Website!