numpy_quaternion_quickstart_creation_array_integration_arithmetic.py

1import numpy as np
2import quaternion
3
4# Quaternions can be created using explicit components
5q1 = np.quaternion(1, 2, 3, 4)
6
7# They integrate seamlessly with numpy arrays
8a = np.array([[1, 2, 3, 4], [5, 6, 7, 8]], dtype=float)
9# You can view an existing array as quaternions (the last dimension must be 4)
10q_array = quaternion.as_quat_array(a)
11
12# Standard arithmetic operations work as expected
13q2 = np.quaternion(1, 0, 0, 0)
14result = q1 * q2 + q1 / 2
15
16# You can also access components directly
17print(f"Quaternion: {q1}")
18print(f"Real part: {q1.w}")
19print(f"Vector part: {q1.vec}")
20print(f"Array of quaternions:\n{q_array}")