Numpy 检查多维数组特定子数组的位置

问题描述 投票:0回答:0

我有 4 维数组,如 (2,48,48,3)。如果最后一个维度不是 [-1,-1,-1],我会向我提供信息,我应该检查该数组位置的非。但我也想高效地做到这一点。我知道如何进行循环,如果有有效的方法请帮助我。顺便说一句,保持尺寸不变对我很有用,就像创建一个面具。

我试过 np.where(data != [-1,-1,-1]),问题是我没看懂这段代码的结果

numpy search numpy-ndarray
© www.soinside.com 2019 - 2024. All rights reserved.