使用`scipy.stats.wasserstein_distance`来计算两个一维分布之间的Earth Mover‘s Distance (EMD)距离

原创

于 2024-03-27 16:41:54 发布 · 783 阅读

CC 4.0 BY-SA版权

文章标签：

在这里插入图片描述

在Python中，计算Earth Mover’s Distance (EMD)通常使用scipy库中的scipy.stats.wasserstein_distance函数，该函数计算的是Wasserstein距离，它与EMD非常相似，都是用来衡量两个分布之间的距离。

以下是一个简单的Python程序例子，展示了如何使用scipy.stats.wasserstein_distance来计算两个一维分布之间的距离：

import numpy as np
from scipy.stats import wasserstein_distance

# 假设我们有两个一维分布，用numpy数组表示它们的概率质量函数
dist_a = np.array([0.1,

200万优质内容无限畅学