将一维数组重塑为蒙版二维数组

Question

好，所以我有这个2D数组：

print indgrid

[[0 6 9 3 1]
[5 9 7 4 5]
[2 8 3 8 4]
[5 6 2 6 8]
[5 3 7 7 0]]

然后，我给它加上一个面具：

print mask

[[False False  True False False]
 [False  True  True  True False]
 [ True  True  True  True  True]
 [False  True  True  True False]
 [False False  True False False]]


print indgrid[mask]

[9 9 7 4 2 8 3 8 4 6 2 6 7]

因此，这是从indgrid数组中取出的一维值数组。 然后，我想对这些值执行一些计算，然后在完成后，将从每个元素计算出的值放入与indgrid数组具有相同形状的indgrid数组中，因此我将indgrid ：

[[ 0 0 f 0 0 ]
[ 0 f f f 0 ]
[ f f f f f ]
[ 0 f f f 0]
[ 0 0 f 0 0 ]]

其中f代表我计算出的浮点数。 最好的方法是什么？ 谢谢。

Answer 1

原谅可怕的变量名：

f_values = some_function_of(indgrid[mask])
result = np.zeros(indgrid.shape)
result[mask] = f_values