vectorize expression containing if statement

Question

SA-W 2023-7-13

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1995828-vectorize-expression-containing-if-statement

评论： the cyclist 2023-7-14

A = [-5 3 2;2 3 4; 3 5 6; -2 8 9];
%compute product of columns in a vectorized fashion
A(:,1).*A(:,2).*A(:,3);
%calculate the above product only for the rows where the entry in the first column is positive
%solution I do not like. A is not needed anymore after the calculation
A(~(A(:,1)>0),:) = [];
A(:,1).*A(:,2).*A(:,3)
ans = 2×1
    24
    90

Is there a way to get the above 2x1 result vector using one single line of code? I would also be happy with a 4x1 vector that contains NaN when the condition is not met. The issue is that my real A is quite large in size, so I want to be as efficient as possible and avoiding unnecessary copies or temporary variables related to the condition.

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

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

Answer 1

the cyclist 2023-7-13

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1995828-vectorize-expression-containing-if-statement#answer_1272908

编辑：the cyclist 2023-7-13

在 MATLAB Online 中打开

Here is one way:

A = [-5 3 2;2 3 4; 3 5 6; -2 8 9];
prod(A(A(:,1)>0,:),2)
ans = 2×1
    24
    90

You could also have done the first operation as

prod(A,2)
ans = 4×1
   -30
    24
    90
  -144

2 个评论
显示无隐藏无

SA-W 2023-7-13

"Product of columns" was actually just an example that I created, so prod() is not really what I am looking for.

In my real code, A has six columns that contain the six independent components of a 3x3 symmetric matrix. I wanna compute the trace of the inverse matrix (formulae known) , but only if the determinant of the matrix is greater than zero. Do you have an idea for that as well?

the cyclist 2023-7-14

在 MATLAB Online 中打开

This question is so different from what you oringally asked, that I'm wary of what is important.

Can't you just do

if det(M) > 0
    <do the trace operation on M here>
end

?

If that isn't right, can I suggest you post a new question (because what you actually asked here was answered). You can upload your matrix using the paper clip icon in the toolbar, and then people can test code for speed on your actual data.

请先登录，再进行评论。