Python克里金插值包pykrige的基本使用

克里金插值简要介绍

克里金(kriging)插值是在有限区域内对区域化变量进行无偏最优估计的一种方法(用于估计在空间上有相关性的值,比如空气质量,相隔很近的位置的数值接近)。无偏指的是估计值和实际值之差的期望等于零,最优指的是估计值和实际值的方差最小。基于这一特点使得克里金插值的效果比其他插值方法要好很多。

简单来说克里金(kriging)插值干的事情就是:已经有一些坐标和这些坐标处的真实值,我们可以称之为采样点。然后把这些采样点输入到克里金插值中,我们就可以用来估计其他未知位置处的值。

在Python里,有两个GitHub评分比较高的克里金插值包,pykrige和pykriging。我两个都用过,感觉pykrige使用起来更加方便好用。

pykrige的基本使用

下面代码给出了使用普通克里金进行插值的一个简单例子,其他类型的克里金插值类似。

from pykrige.ok import OrdinaryKriging
import numpy as np
from matplotlib import pyplot as plt

# 已知采样点的数据,是坐标(x,y)和坐标对应的值
# 矩阵中第一列是x,第二列是y,第三列是坐标对应的值
data = np.array(
    [
        [0.1, 0.1, 0.9],
        [0.2, 0.1, 0.8],
        [0.1, 0.3, 0.9],
        [0.5, 0.4, 0.5],
        [0.3, 0.3, 0.7],
    ])

# 网格
x_range = 0.6
y_range = 0.6
range_step = 0.1
gridx = np.arange(0.0, x_range, range_step) #三个参数的意思:范围0.0 - 0.6 ,每隔0.1划分一个网格
gridy = np.arange(0.0, y_range, range_step)

ok3d = OrdinaryKriging(data[:, 0], data[:, 1], data[:, 2], variogram_model="linear") # 模型
# variogram_model是变差函数模型,pykrige提供 linear, power, gaussian, spherical, exponential, hole-effect几种variogram_model可供选择,默认的为linear模型。
# 使用不同的variogram_model,预测效果是不一样的,应该针对自己的任务选择合适的variogram_model。

k3d1, ss3d = ok3d.execute("grid", gridx, gridy) # k3d1是结果,给出了每个网格点处对应的值


print(np.round(k3d1,2))
#输出的结果
[[0.91 0.87 0.81 0.75 0.7 0.66]
 [0.92 0.9 0.8 0.74 0.68 0.63]
 [0.92 0.89 0.81 0.72 0.65 0.59]
 [0.91 0.9 0.8 0.7 0.62 0.55]
 [0.88 0.84 0.77 0.68 0.59 0.5]
 [0.84 0.8 0.74 0.67 0.59 0.53]]

# 绘图
fig, (ax1) = plt.subplots(1)
ax1.imshow(k3d1, origin="lower")
ax1.set_title("ordinary kriging")
plt.tight_layout()
plt.show()


黄色是值比较大的区域

posted @ 2021-05-13 10:51  DailyYu  阅读(16117)  评论(2编辑  收藏  举报