Spalten auswählen

Bei den meisten Fragen geht es um eine Spalte: das Durchschnittsgewicht, den größten Hund. Setze den Namen der Spalte in eckige Klammern, um sie auszuwählen:

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())

Dies gibt eine Series aus, benannt nach ihrer Spalte, und anschließend das durchschnittliche Gewicht:

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

Für mehrere Spalten gib eine Liste mit Namen in den Klammern an. Du erhältst einen kleineren DataFrame, dessen Spalten in deiner Reihenfolge stehen:

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

Das gibt Folgendes aus:

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

Die Auswahl von Spalten sorgt dafür, dass ein Modell später genau die Merkmale erhält, aus denen es lernen soll.