如何在具有稀疏数据的2D numpy数组上进行线性插值？

Question

我有一个2D numpy数组...图像中有一些值，其余部分稀疏。 对于线性插值，我想获取数组的第一列。 查看存在值的位置，并对零值但仅对一个间隔进行线性插值。

我们遍历二维数组的每一列

例如，考虑以下内容作为第一列

   a = [0,0,0,0,1,0,0,0,2,0,0,10,0,0,3,4,6,0,0,1,0,0]

前四个0,0,0,0将与第一个non_zero元素的副本相同，在本例中为1。

第二个线性插值间隔为

   [1,0,0,0,2]

第三和其余将是

   [2,0,0,10]
   [10,0,0,3]
   [6,0,0,1]

最后，将复制最后一个元素。

非常感谢

Answer 1

尝试这样的事情：

import numpy as np

a = np.array([0,0,0,0,1,0,0,0,2,0,0,10,0,0,3,4,6,0,0,1,0,0])
x, = np.nonzero(a)
a_filled = np.interp(np.arange(a.size), x, a[x])

这样产生：

array([1, 1, 1, 1, 1, 1.25, 1.5, 1.75, 2, 4.67, 7.33, 10, 7.67, 5.33, 3, 4, 6, 4.33, 2.67, 1, 1, 1])