Comment convertir une colonne ou une ligne Pandas DataFrame en liste ?

Comment convertir une colonne ou une ligne Pandas DataFrame en liste

Dans un pandas DataFrame, chaque colonne est représentée par un objet pandas Series. Pour obtenir une représentation de liste d'une colonne, vous pouvez utiliser la méthode tolist() sur l'objet Series. Par exemple :

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

data_dict = {'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 = pd.DataFrame(data_dict)
cluster_list = df['cluster'].tolist()
print(cluster_list)</code>

Sortie :

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

Vous pouvez également convertir un objet Series en une liste directement en utilisant list() :

<code class="python">cluster_list = list(df['cluster'])</code>

Pour obtenir une représentation de liste d'une ligne entière, vous pouvez y accéder en utilisant la méthode iloc() du DataFrame.

<code class="python">row1_list = df.iloc[0].tolist()
print(row1_list)</code>

Sortie :

[1000, '4000', 'Y']

De même, vous pouvez convertir la ligne entière à une liste directement :

<code class="python">row1_list = list(df.iloc[0])</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!