Wie extrahiere ich Daten aus Pandas DataFrames als Listen?

Listen aus Pandas DataFrames abrufen

Bei der Arbeit mit Pandas DataFrames ist es häufig erforderlich, Daten als Listen zur weiteren Analyse oder Verarbeitung zu extrahieren. Hier sind die Antworten auf Ihre Fragen:

Eine Spalte als Liste abrufen

Um eine Spalte als Liste abzurufen, greifen Sie einfach als Pandas-Serie darauf zu und konvertieren Sie sie dann es mit .tolist() in eine Liste einfügen.

import pandas as pd

df = pd.DataFrame({
    'cluster': ['A', 'A', 'A', 'B', 'B', 'B', 'C', 'C', 'C'],
    'load_date': ['1/1/2014', '2/1/2014', '3/1/2014', '4/1/2014', '4/1/2014', '4/1/2014', '7/1/2014', '8/1/2014', '9/1/2014'],
    'budget': [1000, 12000, 36000, 15000, 12000, 90000, 22000, 30000, 53000],
    'actual': [4000, 10000, 2000, 10000, 11500, 11000, 18000, 28960, 51200],
    'fixed_price': ['Y', 'Y', 'Y', 'N', 'N', 'N', 'N', 'N', 'N']
})

df['cluster'].tolist()

Dies gibt eine Liste mit den Werten in der Spalte „Cluster“ zurück:

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

Abrufen einer Zeile oder Spalte als Liste

Um den gesamten Inhalt einer Zeile oder Spalte als Liste abzurufen, verwenden Sie das Attribut .values.

Um beispielsweise Zeile 2 als Liste abzurufen:

df.iloc[1].values.tolist()

Und um Spalte 3 als Liste abzurufen:

df['actual'].values.tolist()

Das obige ist der detaillierte Inhalt vonWie extrahiere ich Daten aus Pandas DataFrames als Listen?. Für weitere Informationen folgen Sie bitte anderen verwandten Artikeln auf der PHP chinesischen Website!