amep.utils.rotation

amep.utils.rotation(theta: float, axis: str = 'z') → ndarray

Rotation matrix in 3D for rotation around a certain axis.

Notes

Only axis=’z’ is implemented in this version.

Parameters:

• theta (float) – rotation angle

• axis (str) – ‘x’, ‘y’, or ‘z’; defines the rotation axis.

Returns:

rotation matrix of shape (3,3)

Return type:

np.ndarray