有没有一种快速的方法将值重新插入到Numpy数组中

Question

好的，我有一些由遮罩s定义的分段图像数据s，它基本上是一个3D二进制字段，其中1定义了属于该部分的体素，0定义了不属于其中的体素。我正在尝试使这一部分的代表尽可能小。这很简单，我可以使用以下代码：

compressed = s.flatten()[m.flatten() == 1]

我的问题是，给定compressed和m，是否有类似的Numpy方法或同样快速的方法来重建s？

Answer 1

感觉很愚蠢，我没有意识到我可以使用以下内容：

a = np.zeros(m.shape)
a[m == 1] = compressed

>>> np.equal(a, s).all()
True

希望这对那些也无法弄清楚的人还是有用的！