takeとtake_whileをitertoolsを使うよう改修