Vectorize 3 for loops

Hello the community !

% let us consider tab, a 3D-matrix, and second_tab, another one
max = 5;

for a_idx = 1:50
	for b_idx = 1:60
		for c_idx = 1:10
			if (tab(a_idx, b_idx, c_idx) > max)
				tab(a_idx, b_idx, c_idx) = Inf;
			else
				tab(a_idx, b_idx, c_idx) = second_tab(a_idx, b_idx, c_idx) * 5;
			end
		end
	end
end

Do you guys know how I could optimize this processus, using Matlab vectorization, (permute, ones, reshape, .*, ./, an. so on.) ?

I can manage similarly cases but with the "if statement" I am blocked....

Robin

Answer 1

[Raghunandan V]

Hi, Here is your answer:

Max = 5;
tab = randi(10,50,60,10);
second_tab =  randi(5,50,60,10);
new_tab = second_tab*5;
new_tab(tab>Max)= inf;

I think this is pretty straightforward. Please reply if explanation is required

[madhan ravi]

Nice approach but don’t name a variable max as it will shadow the in-built function max().

[Raghunandan V]

Yup. I will change it!

[Robin L.]

Raghunandan V I thought something like that existed but I didn't manage to get it works. So thank you so much !