Generate Array In Python Code