def memoryPills(pills): gen = dropwhile(lambda s: len(s) % 2 != 0, pills + [""] * 3) next(gen) return [next(gen) for _ in range(3)]