Построение дерева из списка os путей к файлам (Python) - зависит от производительности

Привет, я работаю над очень высокопроизводительным инструментарием для управления файлами/анализа, написанным на python. Я хочу создать функцию, которая дает мне список или что-то подобное в формате дерева. Что-то вроде этого вопрос (связанный с java)

From:

dir/file
dir/dir2/file2
dir/file3
dir3/file4
dir3/file5

Примечание: список путей не сортирован

To:

dir/
    file
    dir2/
        file2
    file3
dir3/
    file4
    file5

[[dir, [file, [dir2, [file2]], file3]], [dir3, [file4, file5]]]

что-то в этом роде. Я перебрал несколько идей, но ни одна из них не дала той скорости, которую я хотел бы иметь.

Примечание: у меня уже есть список путей, так что об этом можно не беспокоиться. Функция принимает список путей и выдает список деревьев.

Спасибо заранее