Как интерполировать некоторые части массива 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, со значением ближайшего соседнего элемента.

Кто-нибудь может дать некоторые советы? Спасибо.