Selección de columnas

La mayoría de las preguntas se refieren a una columna: el peso promedio, el perro más alto. Pon el nombre de la columna entre corchetes para seleccionarla:

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

Esto imprime una Serie, cuyo nombre es el de su columna, y después el peso promedio:

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

Para varias columnas, coloca una lista de nombres entre corchetes. Obtienes un DataFrame más pequeño, con sus columnas en el orden que indiques:

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

Esto imprime:

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

Seleccionar columnas es la forma en que un modelo obtiene después exactamente las características de las que debe aprender.