将索引中的值分组到

Question

我有一个熊猫数据帧，索引为'assigned'，值为'ticket\u number'：

                    ticket_number
assigned                         
person1                    1
person1                    2
person2                    3

我试着按它们的共同索引对票号进行分组，这样我就可以有唯一的索引。我的目标是将其转换为一对多关系的格式，如下所示：

                    ticket_number
assigned                         
person1                    1
                           2
person2                    3

最终目标是通过to_json（）运行它，以获得以下输出：

{
    "ticket_number": {
        "person1": { "1","2" },
        "person2": "3",
    }
}

似乎to_json()中的一个orientparameters可以适应这种情况，但是没有一个能产生我想要的结果。你知道吗

如何对公共索引中的值进行分组，以便索引只出现一次，并且每个索引项包含关联的值集？你知道吗

Answer 1

听起来您想使用DataFrame.groupby方法。你知道吗

df.groupby(df.index).agg(lambda x: list(x))     

         ticket_number
assigned              
person1         [1, 2]
person2            [3]

然后在整个结果数据帧上，一个Dataframe.to_json()

df.groupby(df.index).agg(lambda x: list(x)).to_json()
'{"ticket_number":{"person1":[1, 2],"person2":[3]}}'

绒布：

{
    "ticket_number": {
        "person1": [1, 2],
        "person2": [3]
    }
}