Define Array Size Python