将矩阵转换为排序矩阵Python

def convert_to_sorted_matrix(matrix):
    rows = len(matrix)
    cols = len(matrix[0])
    
    # 将矩阵的元素取出并存储到一维数组中
    flattened = [matrix[i][j] for i in range(rows) for j in range(cols)]
    
    # 对一维数组进行排序
    sorted_arr = sorted(flattened)
    
    # 重新填充矩阵
    sorted_matrix = [[sorted_arr[i * cols + j] for j in range(cols)] for i in range(rows)]
    
    return sorted_matrix

# 示例矩阵
matrix = [[9, 8, 7],
          [6, 5, 4],
          [3, 2, 1]]

sorted_matrix = convert_to_sorted_matrix(matrix)
print(sorted_matrix)

[[1, 2, 3],
 [4, 5, 6],
 [7, 8, 9]]