How To Define Array Size In Python Iteration