count items matching a rule python