如何基于K近邻计算目标特征（如z值）的平均值-Python教程-PHP中文网

如何基于K近邻计算目标特征（如z值）的平均值

本文详解如何利用scikit-learn的nearestneighbors获取k个最近邻索引，并高效计算每个样本对应邻居的目标维度（如z坐标）的均值，支持任意k值扩展，代码简洁可复用。

本文详解如何利用scikit-learn的nearestneighbors获取k个最近邻索引，并高效计算每个样本对应邻居的目标维度（如z坐标）的均值，支持任意k值扩展，代码简洁可复用。

在空间数据分析、插值建模或局部平滑任务中，常需对每个样本点基于其K个最近邻（KNN）计算某项特征（如高度z、温度、响应值等）的局部平均值。关键在于：KNN返回的是邻居在原始数据集中的索引，而非直接的特征值；因此需通过索引安全提取对应特征并聚合。

以下是一个完整、可扩展的实现方案：

✅ 核心步骤与代码实现

首先，明确数据结构：假设 X 是形状为 (n_samples, n_features) 的二维数组，其中最后一列（索引为 2）代表待平均的 z 值。我们仅用前 n_features−1 列（如 x, y）作距离度量，但需从全量 X 中提取 z。

import numpy as np
from sklearn.neighbors import NearestNeighbors
from statistics import mean  # 更清晰、支持空列表（需注意K≥1）

# 示例数据：每行 = [x, y, z]
X = np.array([
    [6,   -3,  0.1],
    [-5,  -9,  0.5],
    [3,   -7,  0.8],
    [-10,  6,  0.5],
    [-4, -16,  0.9],
    [1, -0.5,  0.0]
])

# Step 1: 构建KNN模型（仅用x,y坐标计算距离）
# 注意：这里使用全部列会将z也纳入距离计算——通常不希望如此
X_coords = X[:, :2]  # 提取前两维作为空间坐标
k = 2
nbrs = NearestNeighbors(n_neighbors=k, metric='euclidean').fit(X_coords)

# Step 2: 获取每个点的K个最近邻索引（含自身？否！默认exclude self）
distances, indices = nbrs.kneighbors(X_coords)
# indices.shape == (n_samples, k) → 每行是该点的k个邻居索引

# Step 3: 提取每个点所有邻居的z值，并计算均值
z_values = X[:, 2]  # 预提取z列，提升效率
neighbor_z_means = [
    mean(z_values[idx_row])  # 直接索引+内置mean，简洁鲁棒
    for idx_row in indices
]

print("每个点的KNN-z均值:", np.round(neighbor_z_means, 3))
# 输出示例: [0.3, 0.65, 0.5, 0.5, 0.45, 0.3]

⚠️ 关键注意事项

坐标与特征分离：务必区分用于距离计算的坐标（X_coords）和用于聚合的目标特征（z_values）。若误将 z 纳入 fit()，会导致距离失真。
是否包含自身：NearestNeighbors 默认返回包括查询点自身的最近邻（即第一个索引常为自身）。若需严格K个其他邻居，应设 n_neighbors = k + 1 并切片 indices[:, 1:]：
```
nbrs = NearestNeighbors(n_neighbors=k+1).fit(X_coords)
_, indices = nbrs.kneighbors(X_coords)
neighbor_z_means = [mean(z_values[idx_row[1:]]) for idx_row in indices]
```

性能优化：对大数据集，避免Python循环。可用NumPy向量化：

# indices: (n, k), z_values: (n,)
# 将indices展平后索引z_values，再reshape求均值
z_matrix = z_values[indices]  # shape: (n, k)
neighbor_z_means = np.mean(z_matrix, axis=1)