创建一个包含来自熊猫数据框的列表列表的字典

Question

我正在尝试创建一个字典，该字典包含熊猫数据框列中基于另一列的值的列表列表，方法是将该列与可迭代引用的各个内容进行比较，如果该可迭代引用中有3个项目，那么我的列表列表将包含3个列表，每个子列表具有与条件适用的数据框中的行数相同的元素

字典中的其他元素只是参考列表和与参考列表大小相同的另一列。

输出字典将提供bokeh-Multiline数据源，但是 我的执行速度很慢，有没有办法提高效率？

execution times

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],
        )