用 Python 对矩阵进行列排序

假设我们有一个矩阵，我们需要按照升序对每一列排序。

因此，如果输入如下：

则输出为：

为此，我们将遵循以下步骤：

• R := 矩阵的行数，C := 矩阵的列数
• res := 与给定矩阵相同大小的矩阵，并用 0 填充
• 对于范围为 0 到 C 的 col，执行
  • values := 将矩阵[col] 的元素作为向量
  • 对于范围为 0 到 R 的 row，执行
    • res[row, col] := 从 values 中删除最后一个元素
• 返回 res

让我们看看以下实现，以获得更好的理解：

示例

在线演示

class Solution:
   def solve(self, matrix):
      R = len(matrix)
      C = len(matrix[0])
      res = [[0] * C for _ in range(R)]
      for col in range(C):
         values = [r[col] for r in matrix]
         values.sort(reverse=True)
         for row in range(R):
            res[row][col] = values.pop()
      return res
ob = Solution()
matrix = [[11, 21, 31],[6, 6, 4],[1, 11, 8]]
print(ob.solve(matrix))

输入

[[11, 21, 31],
[6, 6, 4],
[1, 11, 8]]

输出

[[1, 6, 4],[6, 11, 8],[11, 21, 31]]

Arnab Chakraborty

更新于：9 月 22，2020

2K+ 浏览

开启你的 职业

通过完成课程获得认证

开始吧