Рекурсивно искать значение с помощью поляров? ⇐ Python

[Anonymous]

Я хочу иметь возможность получать менеджер_id каждого менеджера рекурсивно, используя полярный DF, который имеет два столбца:
"employee_id", "manager_1_id"
В pandas этот код был:

Код: Выделить всё

id_index = df.set_index("employee_id")["manager_1_id"]
for i in range(1, 12):
df[f"manager_{str(i + 1)}_id"] = df[f"manager_{str(i)}_id"].map(id_index)

Каждое значение Manager_id также является идентификатором сотрудника, и в конечном итоге мне нужен столбец для каждого менеджера:
"employee_id, Manager_1_id, Manager_2_id, Manager_3_id, ..."
Есть ли хороший способ добиться этого с помощью поляров без запуска фрагмента pandas? Я пытался перебрать некоторые левые соединения, но это не показалось мне хорошим подходом.
Изменить: пример приведен ниже. Необработанные данные имеют два столбца: идентификатор сотрудника (всего много тысяч сотрудников) и идентификатор сотрудника его непосредственного руководителя.

Код: Выделить всё

employee_id | manager_1_id
1       |      3
2       |      5
3       |      4
4       |      5
5       |

Цель — развернуть это на столбцы (от менеджера_1 до менеджера_12)

Код: Выделить всё

employee_id | manager_1_id | manager_2_id | manager_3_id | ...
1              3             4              5
2              5
3              4             5
5              5
5

Надеюсь, это понятно. Сотрудник 1 подчиняется Сотруднику 3, который подчиняется Сотруднику 4, который подчиняется Сотруднику 5. Сотрудник 5 — генеральный директор, который никому не подчиняется.

Подробнее здесь: https://stackoverflow.com/questions/762 ... ith-polars