python - numpy中是否有类似masked_less的masked_max？

Question

相当于什么

masked_arr = np.ma.masked_max(arr, axis=(-1,-2))

获取

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

# or an equivalent with -- and max value in place

来自

[[[ 0.5488135   0.71518937]
  [ 0.60276338  0.54488318]]]

我试过了

np.ma.max(arr, axis=(-1,-2))

但它只返回像 np.max 这样的值，而不是掩码。我该怎么做？

Answer 1

我认为你甚至不需要使用np.ma，你可以这样做

mask = np.isclose(x, x.max((-1, -2), keepdims = True))

记住对 float 使用np.isclose而不是==。然而，

mask = x == x.max((-1, -2), keepdims = True)

如果您有整数或 == 安全的其他类型，则可以使用。