我目前正在创建一个排名算法来对学生之间的关系进行排名。在我的 NxN 矩阵 F 中：F[i, j] 指的是学生 i 和学生 j 之间的关系。值越高，关系越强。
我的问题是这样的。为了创建由 k 个学生组成的集群，我首先选择 F 的 argmax。如果 argmax 返回索引 (i, j)，我将学生 i 和 j 添加到我的集群中。然后我想沿着轴 i 和 j 找到 F 的 argmax，并将两者中较大的一个作为我集群中的下一个学生。然后我沿着集群中每个学生的轴重复这个过程，直到我的集群中有 k 个学生。
我感到困惑的地方： numpy.argmax() 将扁平轴作为要搜索的轴的参数。如何展平轴 i 和 j 以便可以执行 numpy.argmax(F, flattenedAxis)？
我是 Python 的相对初学者，所以这可能有一个简单的答案，我不知道它的存在。提前致谢。