selection.only_water_residues() -- select only water residues

only_water_residues()

This returns a new selection containing only atoms from the current selection in water residues.