python - 二维矩阵中某个轴上的 Numpy.argmax()

Question

我目前正致力于创建一种排名算法来对学生之间的关系进行排名。在我的 NxN 矩阵 F 中: F[i, j] 指的是学生 i 和学生 j 之间的关系。值越高，关系越强。

我的问题是这样的。为了创建一个由 k 个学生组成的集群，我首先选择 F 的 argmax。如果 argmax 返回索引 (i, j)，我将学生 i 和 j 添加到我的集群中。然后我想找到 F 沿轴 i 和 j 的 argmax，并将两者中较大的作为我集群中的下一个学生。然后我沿着集群中每个学生的轴重复这个过程，直到我的集群中有 k 个学生。

我困惑的地方: numpy.argmax() 将扁平轴作为要搜索的轴的参数。如何展平轴 i 和 j 以便我可以执行 numpy.argmax(F, flattenedAxis)？

我是 Python 的初学者，所以这可能有一个我不知道的简单答案。提前致谢。

Answer 1

这将为您提供矩阵 F 中最大元素的索引。

ind = np.unravel_index(np.argmax(F),F.shape)

这将为您提供第 j 列中最大元素的行号。

max_row_in_col_j = np.argmax(F[:,j])

这将为您提供第 i 行中最大元素的列号。

max_col_in_row_i = np.argmax(F[i,:])