How To Create Subarray In Python Iteration