Set Array In Python