Comment puis-je extraire des données d'un Pandas DataFrame sous forme de liste Python ?

Accès aux données des Pandas DataFrames

Pour récupérer les données d'un Pandas DataFrame sous forme de liste, plusieurs méthodes sont disponibles, selon que vous souhaitez accéder à une colonne ou une ligne spécifique.

Accès à une colonne sous forme de liste

Pour obtenir le contenu d'une colonne sous forme de liste, vous pouvez utiliser la syntaxe suivante :

<code class="python">column_list = df['column_name'].tolist()</code>

Cela renverra une liste Python contenant les éléments de la colonne spécifiée.

Accéder à une ligne sous forme de liste

Pour récupérer une ligne sous forme de liste, vous pouvez utiliser les méthodes iloc ou loc, suivies de .tolist() :

<code class="python">row_list = df.iloc[row_index].tolist()  # row index-based
row_list = df.loc[row_label].tolist()  # row label-based</code>

Exemple

Considérez le DataFrame :

<code class="python">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']})</code>

Accès à la colonne sous forme de liste :

<code class="python">cluster_list = df['cluster'].tolist()  # ['A', 'A', 'A', 'B', 'B', 'B', 'C', 'C', 'C']</code>

Accès à la ligne sous forme de liste :

<code class="python">row_list = df.iloc[0].tolist()  # ['A', '1/1/2014', 1000, 4000, 'Y']</code>

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!