python - Numpy - 列之间的平均距离

Question

我有一个二维数组，我想计算每一列之间的平均距离，同时考虑每一行。

例如，如果我有二维数组:

2  2  3
4  2  5
1  5  2

我想计算所有行中第 1 列和第 2 列、所有行中第 1 列和第 3 列以及所有行中第 2 列和第 3 列之间的平均距离。

第 1 列和第 2 列之间的平均距离为 ( |2-2| + |4-2| + |1-5| )/3，等于 2。

是否有一个 numpy 函数可以完成这个？

Answer 1

是这样的吗？

import numpy as np
x = np.array([[2,2,3],[4,2,5],[1,5,2]])

def calc(cols):
    return np.mean(np.abs(np.diff(x[:, cols])))

print(calc([0,1]))

输出:

2.0

还要考虑:

import itertools
print(list(itertools.combinations(range(x.shape[1]), 2)))  # outer list because using py3

输出:

[(0, 1), (0, 2), (1, 2)]