Define Array Size In Python