fill/predict python 中矩阵的未知值

Question

我有一个过程可以在矩阵的某些位置为我提供一些已知值，例如 1x1 米单元格中的人数。因此，如果我们有一个 5x6 米的房间，也许我们只检测到一些单元格中有一定数量的人。

现在我们要创建热图或表面表示，所以有三个选项：

将未知单元格设置为未知
将未知单元格设置为零
尝试预测并填充未知单元格

我想知道如何执行 python 中的第三个选项。我认为这种操作在图像处理（如缩放）中很常见。这个想法是用一组有限的数据使热图平滑。

此致。

Answer 1

假设您不需要 'YES' 矩阵中指定的确切边值（请参阅我的评论），这就是我要做的：

import numpy as np
from scipy.interpolate import griddata

nan = np.NaN
dat = np.array([[ 1,    nan,  nan,  nan,   5,],
                [ nan,  nan,  nan,  nan,  nan,],
                [ nan,  6,    nan,  nan,  8,  ],
                [ nan,  nan,  9,    nan,  nan,],
                [ nan,  nan,  nan,  nan,  nan,],
                [ 0,    nan,  nan,  nan,  2,  ]])

def fill_nans(indata, method='linear'):
    """
    Fill NaN values in the input array `indata`.
    """
    # Find the non-NaN indices
    inds = np.nonzero(~np.isnan(indata))
    # Create an `out_inds` array that contains all of the indices of indata.
    out_inds = np.mgrid[[slice(s) for s in indata.shape]].reshape(indata.ndim, -1).T
    # Perform the interpolation of the non-NaN values to all the indices in the array:
    return griddata(inds, indata[inds], out_inds, method=method).reshape(indata.shape)

out = fill_nans(dat)
print(out)

这给出了，

[[ 1.          2.          3.          4.          5.        ]
 [ 0.8         4.          5.          5.83333333  6.5       ]
 [ 0.6         6.          6.66666667  7.33333333  8.        ]
 [ 0.4         5.25        9.          7.5         6.        ]
 [ 0.2         4.5         5.          5.5         4.        ]
 [ 0.          0.5         1.          1.5         2.        ]]

你也可以，

out2 = fill_nans(dat, method='cubic')
print(out2)

这给出了，

[[ 1.          2.34765155  3.45834401  4.33986447  5.        ]
 [ 2.24368285  4.39570784  5.76911468  6.7162754   6.94217514]
 [ 2.88169911  6.          7.62769189  8.27187136  8.        ]
 [ 2.79787395  6.53998191  9.          8.99319441  7.42165234]
 [ 1.87603253  5.20787111  6.8176744   6.80953373  5.26441632]
 [ 0.          1.73565977  2.59374609  2.65495937  2.        ]]

显然，如果您只想要整数值，您可以在 fill_nans(...) 调用的末尾添加一个 .round().astype(int)，在这种情况下 out 是：

[[1 2 3 4 5]
 [1 4 5 6 6]
 [1 6 7 7 8]
 [0 5 9 8 6]
 [0 4 5 6 4]
 [0 0 1 2 2]]

而out2是：

[[1 2 3 4 5]
 [2 4 6 7 7]
 [3 6 8 8 8]
 [3 7 9 9 7]
 [2 5 7 7 5]
 [0 2 3 3 2]]

同样，显然这些与您的 'YES' 矩阵不完全相同，但希望它对您有所帮助。祝你好运！

fill/predict python 中矩阵的未知值

fill/predict unknown values of matrix in python

python

matplotlib

matrix

heatmap