Python Generator - Array Split