def sorted2(data, key=lambda x: -x):
    new_data = []
    for item in data:
        for item_2 in item:
            new_data.append(item_2)
    new_data.sort(key=key)
    result = data[:]
    i = 0
    for j in range(len(result)):
        result[j] = new_data[i: i + len(result[j])]
        i += len(result[j])
    return result


if __name__ == '__main__':
    print(sorted2([[3], [2, 1]], key=lambda x: x))