python - 如何删除数组中包含值大于 10 的行

Question

 np.array([[ 0,  1,  2,  0,  4,  5,  6,  7,  8,  9],
              [ 0, 11,  0, 13,  0, 15,  0, 17, 18,  0]])

如何删除数组中大于 10 的行？

Answer 1

我们可以通过几个步骤完成此操作。假设 a 是原始数组，那么:

1. 首先我们计算每一行的最大值；
2. 然后我们检查该值是否小于或等于 10；
3. 最后，我们将其用于屏蔽，从而仅选择值不大于 10 的行。

喜欢:

a[a.max(axis=1) <= 10]

因此这将返回原始数组的副本，其中仅包含不包含大于 10 的值的行。