python - 如何插入 numpy 数组的某些部分，它们在最大 k 值中具有最近邻元素值？

Question

我遇到一个问题。现在我有一个 numpy 数组'A'。在 A 中，有一些具有最大 k 值的部分，我想用最近邻项对这些部分进行插值以用于 numpy 数组。这个怎么做？

例子：

A = np.array([[1, 0, 2, 3, 3],
              [0, 1, 4, 5, 3],
              [3, 2, 1, 5, 2],
              [0, 1, -1, 1, 3],
              [1, 1, 1, 2, 3]])

我想用最近邻元素值插入值 >= 4 的元素。

任何人都可以提供一些建议吗？谢谢。

标签： pythonpython-3.xnumpy