Comment convertir une liste en colonne dans Pandas

Vous pouvez utiliser la syntaxe de base suivante pour convertir une liste en colonne dans un DataFrame pandas :

df['new_column'] = pd.Series(some_list)

L’exemple suivant montre comment utiliser cette syntaxe dans la pratique.

Exemple : convertir une liste en colonne dans Pandas

Supposons que nous ayons le DataFrame pandas suivant qui contient des informations sur divers joueurs de basket-ball :

import pandas as pd

#create DataFrame
df = pd.DataFrame({'team': ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H'],
                   'points': [18, 22, 19, 14, 14, 11, 20, 28],
                   'assists': [5, 7, 7, 9, 12, 9, 9, 4],
                   'rebounds': [11, 8, 10, 6, 6, 5, 9, 12]})

#view DataFrame
print(df)

  team  points  assists  rebounds
0    A      18        5        11
1    B      22        7         8
2    C      19        7        10
3    D      14        9         6
4    E      14       12         6
5    F      11        9         5
6    G      20        9         9
7    H      28        4        12

Le code suivant montre comment convertir une liste appelée vols en colonne dans le DataFrame :

#create list
steals = [4, 4, 3, 2, 3, 5, 0, 1]

#convert list to DataFrame column
df['steals'] = pd.Series(steals)

#view updated DataFame
print(df)

  team  points  assists  rebounds  steals
0    A      18        5        11       4
1    B      22        7         8       4
2    C      19        7        10       3
3    D      14        9         6       2
4    E      14       12         6       3
5    F      11        9         5       5
6    G      20        9         9       0
7    H      28        4        12       1

Notez que les vols ont été ajoutés en tant que nouvelle colonne au DataFrame pandas.

Notez que si la liste contient moins d’éléments que le nombre de lignes dans le DataFrame existant, alors les valeurs NaN seront renseignées dans la colonne :

#create list
steals = [4, 4, 3, 2, 3]

#convert list to DataFrame column
df['steals'] = pd.Series(steals)

#view updated DataFame
print(df)

  team  points  assists  rebounds  steals
0    A      18        5        11     4.0
1    B      22        7         8     4.0
2    C      19        7        10     3.0
3    D      14        9         6     2.0
4    E      14       12         6     3.0
5    F      11        9         5     NaN
6    G      20        9         9     NaN
7    H      28        4        12     NaN

Notez que les trois dernières valeurs de la nouvelle colonne de vols sont simplement des valeurs NaN générées par les pandas.

Ressources additionnelles

Les didacticiels suivants expliquent comment effectuer d’autres tâches courantes chez les pandas :

Comment obtenir la valeur d’une cellule à partir de Pandas DataFrame
Comment renommer l’index dans Pandas DataFrame
Comment trier les colonnes par nom dans Pandas