Assign Array In Python