How To Specify Array Size In Python