从查询表内插数据

Question

阅读查询表

LUT = np.genfromtxt('test.out', delimiter=',', dtype=float)
LUT:
    12, 25, 136, 6743
    13, 26, 139, 6786
    14, 27, 142, 6791
    15, 28, 145, 6789

从LUT读取的值如下：

x1, x2, x3 = 12.5, 25.5, 137

读取LUT中每个给定值（3列）中的相邻两个值，我必须对结果进行线性插值（LUT中的第4列）。

给定的值（x1，x2，x3）属于LUT的第一行和第二行之间。基于此如何读取第一行和第二行之间的结果？

Answer 1

给定coords要进行插值的坐标列表，您可以scipy.spatial.cKDTree用来获取表中2个最接近线性插值所必需的条目。下面的代码显示了一个已矢量化的用法示例。

import numpy as np
from scipy.spatial import cKDTree

# inputs
LTU = np.genfromtxt('test.txt', delimiter=',')

coords = ((12.5, 25.5, 137),
          (13.5, 26.5, 141),
          (14.5, 25.5, 144))

# querying and interpolating
xyz = LTU[:, :3]
val = LTU[:, 3]

del LTU # attempt to clean up memory

tree = cKDTree(xyz)
dist, ind = tree.query(coords, k=2)

d1, d2 = dist.T
v1, v2 = val[ind].T
v = (d1)/(d1 + d2)*(v2 - v1) + v1

print(v)
#[ 6758.73909236  6789.16987298  6790.03575996]