Filtrando Linhas

A maioria das consultas mantém apenas algumas linhas: apartamentos com aluguel abaixo de 1000 por mês, pacientes com mais de 60 anos. Uma comparação em uma coluna resulta em True ou False para cada linha, uma máscara:

flats = pd.DataFrame({
    "area": ["North", "Center", "East", "Center", "West"],
    "rooms": [2, 1, 3, 3, 2],
    "rent": [950, 1200, 1100, 1650, 800],
})
print(flats[flats["rent"] < 1000])

Dentro dos colchetes da tabela, a máscara mantém as linhas True, com seus rótulos:

    area  rooms  rent
0  North      2   950
4   West      2   800

Combine condições com & (ambas) ou | (qualquer uma), cada uma entre parênteses, como nas máscaras de array:

print(flats[(flats["rooms"] >= 2) & (flats["rent"] < 1200)])

Isso imprime:

    area  rooms  rent
0  North      2   950
2   East      3  1100
4   West      2   800

O texto também funciona: flats["area"] == "Center". E .shape[0] conta as linhas que corresponderam.