How To Define Array Size In Python For Loops