skrobot.coordinates.math._wrap_axis

skrobot.coordinates.math._wrap_axis(axis)

Convert axis to float vector.

Parameters:

axis (list or numpy.ndarray or str or bool or None) – rotation axis indicated by number or string.

Returns:

axis – conveted axis

Return type:

numpy.ndarray

Examples

>>> from skrobot.coordinates.math import _wrap_axis
>>> _wrap_axis('x')
array([1, 0, 0])
>>> _wrap_axis('y')
array([0, 1, 0])
>>> _wrap_axis('z')
array([0, 0, 1])
>>> _wrap_axis('xy')
array([1, 1, 0])
>>> _wrap_axis([1, 1, 1])
array([1, 1, 1])
>>> _wrap_axis(True)
array([0, 0, 0])
>>> _wrap_axis(False)
array([1, 1, 1])