Update indexing eigenvectors

moble · Mar 12, 2024 · 01f23d6 · 01f23d6
1 parent 9a295d2
commit 01f23d6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/quaternionic/converters.py b/quaternionic/converters.py
@@ -249,15 +249,15 @@ def from_rotation_matrix(cls, rot, nonorthogonal=True):
                 if not shape:
                     q = np.empty((4,), dtype=rot.dtype)
                     eigvals, eigvecs = linalg.eigh(K3.T, subset_by_index=(3, 3))
-                    q[0] = eigvecs[-1]
+                    q[0] = eigvecs[-1, 0]
                     q[1:] = -eigvecs[:-1].flatten()
                     return cls(q)
                 else:
                     q = np.empty(shape+(4,), dtype=rot.dtype)
                     for flat_index in range(reduce(mul, shape)):
                         multi_index = np.unravel_index(flat_index, shape)
                         eigvals, eigvecs = linalg.eigh(K3[multi_index], subset_by_index=(3, 3))
-                        q[multi_index+(0,)] = eigvecs[-1]
+                        q[multi_index+(0,)] = eigvecs[-1, 0]
                         q[multi_index+(slice(1,None),)] = -eigvecs[:-1].flatten()
                     return cls(q)