以下是一个示例代码，用于按照矩阵中元素所在的行和列对矩阵进行排序：

def sort_matrix(matrix):
    rows = len(matrix)
    cols = len(matrix[0])

    # 创建一个包含矩阵中所有元素及其行列信息的列表
    elements = []
    for i in range(rows):
        for j in range(cols):
            elements.append((matrix[i][j], i, j))

    # 按照元素的行列信息对列表进行排序
    elements.sort(key=lambda x: (x[1], x[2]))

    # 重新构建排序后的矩阵
    sorted_matrix = [[0] * cols for _ in range(rows)]
    for i in range(rows):
        for j in range(cols):
            sorted_matrix[i][j] = elements[i * cols + j][0]

    return sorted_matrix

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

# 调用排序函数并输出结果
sorted_matrix = sort_matrix(matrix)
print(sorted_matrix)

输出:

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

这段代码首先创建了一个包含矩阵中所有元素及其行列信息的列表 elements。然后，通过使用元组的第二个和第三个元素作为排序键值，对 elements 列表进行排序。最后，根据排序后的 elements 列表重新构建排序后的矩阵 sorted_matrix。