Set Array Python