Vectorization: incompatible sizes of arrays

4 Ansichten (letzte 30 Tage)
Milan Lstiburek
Milan Lstiburek am 18 Sep. 2022
Kommentiert: Torsten am 19 Sep. 2022
O is a matrix with n_p x n_o rows; P is a matrix with n_p rows. Both matrices have three columns.The second column in O should be filled with the average of two values in the second column of P: the first value at the row ip and the second value at randomly selected row.
counter = 0;
for ip = 1:n_p
for io = 1:n_o
counter = counter + 1;
O(counter,2) = (P(ip,2) + P(randi(n_p),2))/2;
end
end
I attempted to vectorize the code as:
ip = 1:n_p;
io = 1:n_o;
O(ip(:)+io(:),2) = (P(ip(:),2)+P(randi(n_p),2))/2;
I am receiving the following message: „Arrays have incompatible sizes for this operation“. Thank you for any suggestions.

Akzeptierte Antwort

Torsten
Torsten am 18 Sep. 2022
Bearbeitet: Torsten am 18 Sep. 2022
n_p = 3;
n_o = 3;
P = rand(n_p,2);
R = randi(n_p,n_p*n_o,1)
R = 9×1
3 2 2 2 3 1 3 2 1
IP = (cell2mat(arrayfun(@(i)vertcat(repmat(i,1,n_o)),1:n_p,'UniformOutput',0))).'
IP = 9×1
1 1 1 2 2 2 3 3 3
O(:,2) = (P(IP,2)+P(R,2))/2
O = 9×2
0 0.8950 0 0.5663 0 0.5663 0 0.2625 0 0.5912 0 0.5663 0 0.9199 0 0.5912 0 0.8950
  3 Kommentare
Bruno Luong
Bruno Luong am 19 Sep. 2022
Bearbeitet: Bruno Luong am 19 Sep. 2022
IP = (cell2mat(arrayfun(@(i)vertcat(repmat(i,1,n_o)),1:n_p,'UniformOutput',0))).'
can be simplified to
IP = replem((1:n_p)',n_o)
Torsten
Torsten am 19 Sep. 2022
Or:
IP = reshape(repmat(1:n_p,n_o,1),n_p*n_o,1)
But yours is shorter :-)

Melden Sie sich an, um zu kommentieren.

Weitere Antworten (2)

Bruno Luong
Bruno Luong am 19 Sep. 2022
Bearbeitet: Bruno Luong am 19 Sep. 2022
Simpler one liner
O(:,2) = P(randi(end,n_o,end),2) + repelem(P(:,2),n_o) / 2;

Bruno Luong
Bruno Luong am 18 Sep. 2022
Bearbeitet: Bruno Luong am 18 Sep. 2022
O(:,2) = reshape(reshape(P(randi(end,n_o,end),2),n_o,[]) + reshape(P(:,2),1,[]), [],1) / 2;
  2 Kommentare
Milan Lstiburek
Milan Lstiburek am 18 Sep. 2022
I just checked and it seems to work equally. I like your approach very much, never thought this could be done in a single line. Thank you!
Bruno Luong
Bruno Luong am 19 Sep. 2022
Good.
It is not easy to check the correctness due to randomness.

Melden Sie sich an, um zu kommentieren.

Produkte


Version

R2022b

Community Treasure Hunt

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

Start Hunting!

Translated by