Comment accéder aux colonnes et aux lignes DataFrame sous forme de listes dans la bibliothèque Pandas de Python ?

Accès aux colonnes et aux lignes du DataFrame sous forme de listes

Dans la bibliothèque Pandas de Python, un DataFrame contient des lignes et des colonnes de données tabulaires. Pour accéder au contenu d'une colonne ou d'une ligne DataFrame, vous pouvez utiliser les méthodes suivantes :

1. Obtenir le contenu d'une colonne

Pour récupérer le contenu d'une colonne DataFrame sous forme de liste, utilisez la méthode tolist() sur l'objet Series représentant la colonne. Vous pouvez également diffuser la série dans une liste à l'aide de la fonction list().

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

# Create a DataFrame from sample data
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']
})

# Convert column values to a list
cluster_list = df['cluster'].tolist()

# Alternatively, you can cast the Series to a list
cluster_list = list(df['cluster'])</code>

2. Obtenir le contenu des lignes

Pour obtenir le contenu d'une ligne DataFrame sous forme de liste, utilisez l'accesseur loc ou iloc avec l'index de ligne approprié.

<code class="python"># Get row 1 as a list using 'loc'
row_1_list = df.loc[0].tolist()

# Get row 1 as a list using 'iloc'
row_1_list = df.iloc[0].tolist()</code>

Exemple de sortie :

cluster_list: ['A', 'A', 'A', 'B', 'B', 'B', 'C', 'C', 'C']
row_1_list: [1000, 4000, 'Y']

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!