Я пытаюсь создать словарь, содержащий списки списков из столбца кадра данных pandas на основе значения другого столбца, сравнивая его с индивидуальным содержимым ссылочного итеративного объекта, если в указанном итерируемый, то мой список списков будет содержать 3 списка, каждый подсписок с таким количеством элементов, сколько строк в кадре данных, где применяется условие
другие элементы в словаре - это просто список ссылок, а другой столбец того же размера, что и список ссылок.
Выходной словарь будет передавать многострочный источник данных боке, но моя реализация очень медленная, есть ли способ сделать его более эффективным?
время выполнения
leg_list = df2['main_legend'].unique().tolist()
lin_dict = dict(
X = [df2.loc[df2['main_legend'] == i, "X"].tolist() for i in leg_list],
Y = [df2.loc[df2['main_legend'] == i, "Y"].tolist() for i in leg_list],
line_legend = leg_list,
color = [df2.loc[df2['main_legend'] == i, "color"].values[0] for i in leg_list],
)