numpy_quaternion_array_arithmetic_and_exp_log_operations.py

1import numpy as np
2import quaternion
3
4# Create a single quaternion
5q1 = np.quaternion(1, 2, 3, 4)
6
7# Create an array of quaternions
8# Note: The dtype must be specified as np.quaternion
9a = np.array([[1, 2, 3, 4], [5, 6, 7, 8]], dtype=np.quaternion)
10
11# Standard arithmetic operations work element-wise
12q2 = q1 * q1
13q3 = q1 + q1
14
15# Use built-in numpy functions
16# Note: Many functions like exp, log, and sqrt are supported
17q_exp = np.exp(q1)
18
19# Quaternion-specific properties
20print(f"Quaternion: {q1}")
21print(f"Real part: {q1.real}")
22print(f"Imaginary part: {q1.imag}")
23print(f"Norm: {np.abs(q1)}")