Comment convertir une colonne ou une ligne Pandas DataFrame en liste ?-Tutoriel Python-php.cn

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

Mary-Kate Olsen

Libérer： 2024-10-27 01:02:30

original

996 Les gens l'ont consulté

How do you convert a Pandas DataFrame column or row to a list?

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>

Copier après la connexion

Sortie :

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

Copier après la connexion

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

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

Copier après la connexion

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>

Copier après la connexion

Sortie :

[1000, '4000', 'Y']

Copier après la connexion

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

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

Copier après la connexion

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!