使用gdal均匀筛选点矢量

使用gdal均匀筛选点矢量

作用：

通过计算各点之间的欧式距离，筛选出符合目标的、均匀发布在空间中的N个数据点。

效果示意图

运行环境

python 3.10
安装：tqdm、numpy和tqdm这三个库

完整代码

import numpy as np
from osgeo import ogr, osr
from tqdm import tqdm

# 代码作用：通过计算各点之间的欧式距离，筛选出符合目标的、均匀发布在空间中的N个数据点。



# 定义需要采样的个数
n_samples = 100
input_path = r"测试数据\村点.shp"
output_path = r"测试数据\samples.shp"

# 1. 读取原始点数据
driver = ogr.GetDriverByName('ESRI Shapefile')
inds = driver.Open(input_path, 0)
layer = inds.GetLayer()

# 2. 提取点坐标和属性
coords = []
attrs = []
for feature in layer:
    geom = feature.GetGeometryRef()
    coords.append((geom.GetX(), geom.GetY()))
    attrs.append([feature.GetField(i) for i in range(feature.GetFieldCount())])
coords = np.array(coords)
attrs = np.array(attrs)

# 3. 定义距离函数
def distance(p1, p2):
    return np.sqrt(np.sum((p1 - p2)**2))

# 4. 随机选择第一个点
idx = np.random.choice(coords.shape[0], 1)
samples = coords[idx]
sample_attrs = attrs[idx]

# 5. 选择空间均衡的采样点
for _ in tqdm(range(n_samples - 1)):
    dists = np.array([np.min(np.array([distance(p, s) for s in samples])) for p in coords])
    idx = np.argmax(dists)
    samples = np.append(samples, [coords[idx]], axis=0)
    sample_attrs = np.append(sample_attrs, [attrs[idx]], axis=0)
    coords = np.delete(coords, idx, axis=0)
    attrs = np.delete(attrs, idx, axis=0)

# 6. 将采样点转为gdal几何对象
out_samples = []
for sample in samples:
    point = ogr.Geometry(ogr.wkbPoint)
    point.AddPoint(sample[0], sample[1])
    out_samples.append(point)

# 7. 创建新的矢量层并写入采样点
out_driver = ogr.GetDriverByName('ESRI Shapefile')
out_ds = out_driver.CreateDataSource(output_path)
out_layer = out_ds.CreateLayer('samples', layer.GetSpatialRef(), ogr.wkbPoint)

# 添加属性字段
for i in range(len(layer.schema)):
    field_defn = layer.schema[i]
    out_layer.CreateField(field_defn)

# 写入采样点要素
for i, sample in enumerate(out_samples):
    feature = ogr.Feature(out_layer.GetLayerDefn())
    feature.SetGeometry(sample)
    for j, attr in enumerate(sample_attrs[i]):
        feature.SetField(j, attr)
    out_layer.CreateFeature(feature)

out_layer = None
out_ds = None