Sélection de colonnes

La plupart des questions portent sur une colonne : le poids moyen, le chien le plus grand. Mets le nom de la colonne entre crochets pour la sélectionner :

dogs = pd.DataFrame({
    "name": ["Max", "Luna", "Bo", "Kira"],
    "kg": [30, 8, 22, 12],
    "cm": [61, 30, 55, 40],
})
print(dogs["kg"])
print(dogs["kg"].mean())

Cela affiche une Series, nommée d’après sa colonne, puis le poids moyen :

0    30
1     8
2    22
3    12
Name: kg, dtype: int64
18.0

Pour plusieurs colonnes, placez une liste de noms entre les crochets. Vous obtenez un DataFrame plus petit, avec ses colonnes dans l’ordre que vous avez choisi :

print(dogs[["name", "cm"]])

Cela affiche :

   name  cm
0   Max  61
1  Luna  30
2    Bo  55
3  Kira  40

Sélectionner des colonnes permet au modèle d'obtenir ensuite exactement les caractéristiques à partir desquelles il doit apprendre.