如何识别numpy数组中平均值最大的列

Question

给定以下数组：

    complete_matrix = numpy.array([
    [0, 1, 2, 4],
    [1, 0, 3, 5],
    [2, 3, 0, 6]])

我想确定平均值最高的列，不包括对角零。因此，在这种情况下，我可以将complete_matrix[：，3]识别为具有最高平均值的列。

Answer 1

这个问题与这里的问题不同吗：在numpy数组中查找平均值最高的行

据我所知，唯一的区别是这篇文章中的矩阵不是正方形矩阵。如果这是故意的，你可以尝试使用权重。由于我不完全理解您的意图，以下解决方案为零条目分配0权重，否则为1：

numpy.argmax(numpy.average(complete_matrix,axis=0, weights=complete_matrix!=0))

您总是可以创建一个权重矩阵，其中对角线条目的权重为0，否则为1。

Answer 2

类似于：

import numpy
complete_matrix = numpy.array([
    [0, 1, 2, 4],
    [1, 0, 3, 5],
    [2, 3, 0, 6]])
print complete_matrix[:,numpy.argmax(numpy.mean(complete_matrix, 0))]
# [4 5 6]