如何使用numpy获得第n个最大相关系数及其指数？

Question

我计算这样的相关系数（它只是一个例子）：

a = np.array([[1, 2, 3],

     [4, 7, 9],

     [8, 7, 5]])

corr = np.corrcoef(a)

结果是一个相关矩阵。

问题是如何获得第一个、第二个（或第 n 个）最大系数？

以及它的索引？喜欢[0,1]和[2,1]

Answer 1

假设您有一个 NumPy 数组，并且您像这样计算相关系数：import numpy as npa = np.array([[1, 2, 3],     [4, 7, 9],     [8, 7, 5]])corr = np.corrcoef(a)现在展平数组，获取唯一系数并对展平的数组进行排序：flat=corr.flatten()flat = np.unique(flat)平面数组如下所示：>> array([-0.98198051, -0.95382097,  0.99339927,  1. ])现在选择nth largest元素，只需选择正确的索引：largest = flat[-1]second_largest = flat[-2]print(largest)print(second_largest)>> 1.0>> 0.9933992677987828要找到相应系数的索引：result = np.where(corr == largest)indices = np.array(result)print(indices)这将打印出以下数组。因此，出现最大系数的索引是 (0,0)、(1,1) 和 (2,2)。>> array([[0, 1, 2],       [0, 1, 2]])