Create Fixed Array Size Python Code