屏蔽2D阵列并基于屏蔽索引在第二阵列上操作

I have a function that reads in and outputs a 2D array. I want the output to be constant (pi in this case) for every index in the input that equals 0, otherwise I perform some maths on it. E.g:I have a function that reads in and outputs a 2