SciPy

scipy.spatial.transform.Rotation.__getitem__

Rotation.__getitem__(indexer)[source]

Extract rotation(s) at given index(es) from object.

Create a new Rotation instance containing a subset of rotations stored in this object.

Parameters:
indexer : index, slice, or index array

Specifies which rotation(s) to extract. A single indexer must be specified, i.e. as if indexing a 1 dimensional array or list.

Returns:
rotation : Rotation instance
Contains
  • a single rotation, if indexer is a single index
  • a stack of rotation(s), if indexer is a slice, or and index array.

Examples

>>> from scipy.spatial.transform import Rotation as R
>>> r = R.from_quat([
... [1, 1, 0, 0],
... [0, 1, 0, 1],
... [1, 1, -1, 0]])
>>> r.as_quat()
array([[ 0.70710678,  0.70710678,  0.        ,  0.        ],
       [ 0.        ,  0.70710678,  0.        ,  0.70710678],
       [ 0.57735027,  0.57735027, -0.57735027,  0.        ]])

Indexing using a single index:

>>> p = r[0]
>>> p.as_quat()
array([0.70710678, 0.70710678, 0.        , 0.        ])

Array slicing:

>>> q = r[1:3]
>>> q.as_quat()
array([[ 0.        ,  0.70710678,  0.        ,  0.70710678],
       [ 0.57735027,  0.57735027, -0.57735027,  0.        ]])