python - 从numpy中的张量中过滤出零矩阵

Question

我得到了一个大小为 ML 算法的分割掩码张量(30, 256, 256)。问题是，其中一些条目是零矩阵，我必须将它们过滤掉。现在我正在使用带有 np.array_equal 函数的基于循环的简单技术来手动过滤掉它们。

有没有办法使用一些花哨的索引以 NumPy 的方式更有效地做到这一点？

标签： pythonnumpy

Answer 1

如果a是你的阵列，你可以做

a[np.any(a != 0, (1, 2))]