Pythonitertools.takewhile