Alternative array indexing

billtubbs · billtubbs · commit 39604464f02f · 2021-07-31T18:18:36.000-04:00
diff --git a/examples/pvtol-lqr.py b/examples/pvtol-lqr.py
@@ -92,12 +92,12 @@
 alt = (1, 4)
 
 # Decoupled dynamics
-Ax = (A[lat, :])[:, lat]  # ! not sure why I have to do it this way
+Ax = A[np.ix_(lat, lat)]
 Bx = B[lat, 0]
 Cx = C[0, lat]
 Dx = D[0, 0]
 
-Ay = (A[alt, :])[:, alt]  # ! not sure why I have to do it this way
+Ay = A[np.ix_(alt, alt)]
 By = B[alt, 1]
 Cy = C[1, alt]
 Dy = D[1, 1]
