Selecionando colunas

A maioria das perguntas é sobre uma coluna: o peso médio, o cachorro mais alto. Coloque o nome da coluna entre colchetes para selecioná-la:

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

Isso imprime uma Series, nomeada de acordo com sua coluna, e depois o peso médio:

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

Para várias colunas, coloque uma lista de nomes entre colchetes. Você obtém um DataFrame menor, com as colunas na ordem escolhida:

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

Isso imprime:

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

Selecionar colunas é como um modelo obtém exatamente os recursos dos quais deve aprender.