Itertools Count Python Method Parameter