How To Define Array Size In Python