选择具有最大总和 - 矩阵的列的标题值

Select column header value for a column with the largest Sum - Matrix

提问人:Meana 提问时间:11/11/2023 更新时间:11/11/2023 访问量:36

问:

我有一个矩阵,我想为其选择计算出最大总和的列标题值。

matrix= [['a','b','c'],                                                                
          [3, 7, 6],                                                                   
          [1, 3, 5],                                                                   
          [9, 4, 2]]

我计算这个矩阵的总和,我现在要选择列标题值,即具有最大列总和。[13, 14, 13]b

Python 列表 矩阵 数据结构

评论


答:

0赞 Ömer Sezer 11/11/2023 #1

转置、求和并找到最大列。

法典:

matrix = [['a', 'b', 'c'],
      [3, 7, 6],
      [1, 3, 5],
      [9, 4, 2]]

transposed_matrix = list(zip(*matrix[1:]))  
column_sums = [sum(col) for col in transposed_matrix]  # sum of columns
max_sum_index = column_sums.index(max(column_sums))
max_sum_column_header = matrix[0][max_sum_index]
print("Column header:", max_sum_column_header)

输出:

Column header: b

相同代码的几行:

column_sums = [sum(col) for col in zip(*matrix[1:])]
max_sum_column_header = matrix[0][column_sums.index(max(column_sums))]
print("Column header:", max_sum_column_header)