成对距离最小值索引#

sklearn.metrics.pairwise_distances_argmin(X, Y, *, axis=1, metric='euclidean', metric_kwargs=None)[source]#

计算一点与一组点之间的最小距离。

此函数计算 X 中每一行与 Y 中距离最近的行（根据指定的距离）的索引。

这与调用以下函数几乎等效：

pairwise_distances(X, Y=Y, metric=metric).argmin(axis=axis)

但它使用了更少的内存，并且对于大型数组速度更快。

此函数仅适用于密集的二维数组。

参数：
X{array-like, sparse matrix} of shape (n_samples_X, n_features)

包含点的数组。

Y{array-like, sparse matrix} 形状为 (n_samples_Y, n_features)

包含点的数组。

axisint, 默认值=1

计算 argmin 和距离的轴。

metricstr 或 callable，默认值=”euclidean”

用于距离计算的度量。可以使用 scikit-learn 或 scipy.spatial.distance 中的任何度量。

如果 metric 是一个可调用函数，则它会在每一对实例（行）上调用，并将结果值记录下来。可调用函数应接收两个数组作为输入，并返回一个值表示它们之间的距离。这适用于 Scipy 的度量，但效率低于将度量名称作为字符串传递。

不支持距离矩阵。

metric 的有效值为：

  • 来自 scikit-learn：[‘cityblock’, ‘cosine’, ‘euclidean’, ‘l1’, ‘l2’, ‘manhattan’, ‘nan_euclidean’]

  • 来自 scipy.spatial.distance：[‘braycurtis’, ‘canberra’, ‘chebyshev’, ‘correlation’, ‘dice’, ‘hamming’, ‘jaccard’, ‘kulsinski’, ‘mahalanobis’, ‘minkowski’, ‘rogerstanimoto’, ‘russellrao’, ‘seuclidean’, ‘sokalmichener’, ‘sokalsneath’, ‘sqeuclidean’, ‘yule’]

有关这些度量的详细信息，请参阅 scipy.spatial.distance 的文档。

注意

'kulsinski' 已在 SciPy 1.9 中弃用，并将从 SciPy 1.11 中删除。

注意

'matching' 已在 SciPy 1.9 中删除（请改用 'hamming'）。

metric_kwargsdict, 默认值=None

传递给指定度量函数的关键字参数。

返回:
argminnumpy.ndarray

Y[argmin[i], :] 是 Y 中与 X[i, :] 最接近的行。

另请参见

成对距离

X 和 Y 的每对样本之间的距离。

成对距离最小值索引和最小值

与 pairwise_distances_argmin 相同，但还会返回距离。

示例

>>> from sklearn.metrics.pairwise import pairwise_distances_argmin
>>> X = [[0, 0, 0], [1, 1, 1]]
>>> Y = [[1, 0, 0], [1, 1, 0]]
>>> pairwise_distances_argmin(X, Y)
array([0, 1])