Comment extraire les données d'un Pandas DataFrame sous forme de liste ?

Extraire des données d'un Pandas DataFrame sous forme de liste

Dans un Pandas DataFrame, chaque colonne représente une série Pandas, qui peut être convertie en liste Python. Cette capacité permet une itération efficace dans les données et permet un traitement et une visualisation ultérieurs.

Pour extraire le contenu d'une colonne sous forme de liste, la méthode .tolist() ou la conversion list(x) peuvent être utilisées. Par exemple, si vous disposez d'un DataFrame df avec une colonne nommée 'cluster', vous pouvez récupérer son contenu sous forme de liste en utilisant le code suivant :

column1_list = df['cluster'].tolist()

Alternativement, si vous souhaitez obtenir la colonne entière ou ligne sous forme de liste, vous pouvez utiliser la méthode .to_numpy(), qui convertit les données en un tableau NumPy, ou la propriété .values(), qui renvoie un tableau NumPy 2D représentant les données du DataFrame.

Pour Par exemple, pour extraire l'intégralité de la première colonne sous forme de liste, vous utiliseriez :

column1_list = df[df.columns[0]].tolist()

Ou, pour récupérer la première ligne sous forme de tableau :

row1_array = df.iloc[0].to_numpy() # or df.iloc[0].values

En tirant parti de ces techniques, vous pouvez facilement extraire des données des Pandas DataFrames dans des listes pour un traitement ou une analyse ultérieure.

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!