NumPy: умножение матрицы, когда координаты задаются отдельно, есть ли ярлык?

Я работаю с позициями частиц в разных массивах, что-то вроде (три частицы):

import numpy as np

x = np.array([1, 2, 3])
y = np.array([2, 6, 1])
z = np.array([0, 3, 6])

Я делаю это таким образом из-за 3D-графиков matplotlib, которые требуют наличия разных координат в разных массивах (например, ax.scatter(x, y, z)).

Но недостатком этого подхода является то, что если я хочу перемножить матрицы:

mat = np.array([[10, 0, 0], [0, 10, 0], [0, 0, 10]])

Я не могу использовать np.dot напрямую, потому что координаты находятся в разных массивах, поэтому мне нужно объединить их и разъединить после умножения. Я знаю, как это сделать (np.transpose([x, y, z])), но меня беспокоит эффективность, я не знаю, сложны ли эти операции для больших массивов (я работаю с сотнями тысяч или миллионами частиц). Возможно, для этого есть функция numpy, о которой я не знаю. Конечно, реализовать это самостоятельно не составит труда.

Вы можете избежать проблемы XY, используя ax.scatter(*xyz.T) вместо xyz (N, 3).

Timeless 28.05.2024 10:02
Почему в Python есть оператор "pass"?
Почему в Python есть оператор "pass"?
Оператор pass в Python - это простая концепция, которую могут быстро освоить даже новички без опыта программирования.
Некоторые методы, о которых вы не знали, что они существуют в Python
Некоторые методы, о которых вы не знали, что они существуют в Python
Python - самый известный и самый простой в изучении язык в наши дни. Имея широкий спектр применения в области машинного обучения, Data Science,...
Основы Python Часть I
Основы Python Часть I
Вы когда-нибудь задумывались, почему в программах на Python вы видите приведенный ниже код?
LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа
LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа
Алиса и Боб имеют неориентированный граф из n узлов и трех типов ребер:
Оптимизация кода с помощью тернарного оператора Python
Оптимизация кода с помощью тернарного оператора Python
И последнее, что мы хотели бы показать вам, прежде чем двигаться дальше, это
Советы по эффективной веб-разработке с помощью Python
Советы по эффективной веб-разработке с помощью Python
Как веб-разработчик, Python может стать мощным инструментом для создания эффективных и масштабируемых веб-приложений.
0
1
55
1
Перейти к ответу Данный вопрос помечен как решенный

Ответы 1

Ответ принят как подходящий

Я предполагаю, что вам нужно отображать свои точки и/или результаты всего несколько раз в рамках всей вашей системы. По крайней мере, так обычно бывает по моему собственному опыту. Учитывая это, вы можете выполнить все свои вычисления на одной матрице, содержащей все ваши точки, поскольку это НАМНОГО более эффективно. Позже, когда вам понадобится что-то построить, вы можете разделить матрицу на отдельные точки и продолжить. Поскольку построение графиков происходит гораздо реже по сравнению с вычислениями, этими затратами можно пренебречь.

Другие вопросы по теме