Как применить вращения на структурированные массивы Numpy? ⇐ Python

[Anonymous]

Я использую структурированные массивы для хранения данных атомов, полученных Lammps (я использую структурированный массив, который следует за его форматом). Мне нужно повернуть позиции: < /p>

Код: Выделить всё

import numpy as np

transform = np.array([[1, 0, 0], [0, 1, 0], [0, 0, 1]], dtype=np.float64)
dtype = np.dtype([("x", np.float64), ("y", np.float64), ("z", np.float64)])

atoms = np.array(
[
(0.0, 0.0, 0.0),
(1.0, 0.0, 0.0),
(0.0, 1.0, 0.0),
(1.0, 1.0, 1.0),
],
dtype=dtype,
)

atoms[["x", "y", "z"]] = atoms[["x", "y", "z"]] @ transform.T
< /code>
Но это создает: < /p>
Traceback (most recent call last):
File "c:\Users\acgc99\Desktop\rotation.py", line 16, in 
atoms[["x", "y", "z"]] = atoms[["x", "y", "z"]] @ transform.T
~~~~~~~~~~~~~~~~~~~~~~~^~~~~~~~~~~~~
numpy._core._exceptions._UFuncNoLoopError: ufunc 'matmul' did not contain a loop with signature matching types (dtype([('x', '

Подробнее здесь: [url]https://stackoverflow.com/questions/79663073/how-to-apply-rotations-to-structured-numpy-arrays[/url]