Comment extraire des données des DataFrames Pandas sous forme de listes ?

Obtention de listes à partir de Pandas DataFrames

Lorsque vous travaillez avec Pandas DataFrames, il est souvent nécessaire d'extraire des données sous forme de listes pour une analyse ou un traitement plus approfondi. Voici les réponses à vos questions :

Récupérer une colonne sous forme de liste

Pour récupérer une colonne sous forme de liste, accédez-y simplement en tant que série Pandas puis convertissez-la dans une liste en utilisant .tolist().

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()

Cela renverra une liste contenant les valeurs dans la colonne 'cluster' :

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

Récupération d'une ligne ou Colonne sous forme de liste

Pour récupérer l'intégralité du contenu d'une ligne ou d'une colonne sous forme de liste, utilisez l'attribut .values.

Par exemple, pour récupérer la ligne 2 sous forme de liste :

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

Et pour récupérer la colonne 3 sous forme de liste :

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

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!