How to Vectorize this Code?

1 次查看（过去 30 天）
Krishan Bansal
Krishan Bansal 2022-9-21
评论： Krishan Bansal 2022-9-21
I have this 147000 x 3 array, where each index represents a 3D vector. I want to use the vector at each index to find the angle between it and a predefined vector "b0" [ 0 1 0], this results in a 147000 x 1 array where each index has a single angle. I have been able to do this by using a normal for loop, however I was wondering if there was a way to vectorize this so it will be faster. I have been trying, but it either says that it can't do the dot product because arrays are of incompatible size, or it will only calculate the angle of the first index, then copy that 147000 times into the Nx1 array.
Thanks for any help.
  4 个评论
显示 2更早的评论
Matt J
Matt J 2022-9-21
@Krishan Bansal Please post your code and output as text, not images, to facilitate copy/pasting.
Krishan Bansal
Krishan Bansal 2022-9-21
@Matt J ok the code is just:
n3Array = squeeze(Orientation4D(:, 1, 1,1:3));
N = length(n3Array);
b0angles = zeros(N,1);
for i = 1:N
b0angles(i) = rad2deg(acos(dot(n3Array(i,:), b0)/(norm(n3Array(i,:)) * norm(b0))));
end

请先登录，再进行评论。

请先登录，再回答此问题。

采纳的回答

Matt J
Matt J 2022-9-21
编辑：Matt J 2022-9-21
n3Array=normalize(n3Array,2,'n');
b0=b0/norm(b0);
b0angles=acosd(n3Array*b0);
  11 个评论
显示 9更早的评论
James Tursa
James Tursa 2022-9-21
This creates complex doubles?
n3Array = squeeze(Orientation4D(:, 1, 1,1:3));
b0 = [0 1 0];
n3Array = normalize(n3Array,2,'n');
b0 = b0/norm(b0);
b0angles = acosd(n3Array*b0(:));
Krishan Bansal
Krishan Bansal 2022-9-21
@James Tursa That worked perfectly, I forgot to use the normalized version of n3 I think. Thanks so much! This is super concise and efficient code.

请先登录，再进行评论。

更多回答（0 个）

类别

在 Help Center 和 File Exchange 中查找有关 Logical 的更多信息

标签

产品

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by