Выбор столбцов

Большинство вопросов касается одного столбца: среднего веса, самой высокой собаки. Чтобы выбрать столбец, укажи его имя в квадратных скобках:

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

Это выводит Series с именем, совпадающим с именем столбца, а затем средний вес:

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

Для нескольких столбцов поместите список названий в скобки. Вы получите DataFrame поменьше, столбцы которого будут расположены в указанном вами порядке:

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

Это выводит:

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

Выбор столбцов позволяет модели впоследствии получить именно те признаки, на которых ей следует обучаться.