Wybieranie kolumn

Większość pytań dotyczy jednej kolumny: średniej wagi, najwyższego psa. Wpisz nazwę kolumny w nawiasach kwadratowych, aby ją wybrać:

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

To wyświetla serię o nazwie takiej jak jej kolumna, a następnie średnią wagę:

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

W przypadku kilku kolumn umieść listę nazw w nawiasach. Otrzymasz mniejszy obiekt DataFrame z kolumnami w wybranej przez Ciebie kolejności:

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

To wyświetla:

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

Wybieranie kolumn pozwala modelowi później uzyskać dokładnie te cechy, na podstawie których powinien się uczyć.