捕获列值的实例，其中它不止一次地被看到。

Question

我有以下数据框架：

id_1   id_2 id_3    id_4    id_5    
0133    11   kelly  AA-1    1        
2119    22   Wade   AA-2    1        
3903    33   John   BB-1    1        
3903    33   John   BB-2    1        
3903    33   John   BB-3    1        
5133    44   Emily  C-1     1               
9148    99   Pete   BB-34   1       
9148    99   Pete   BB-23   1        
2910    111  Mark   DD-3    1                

我想迭代它并捕获当id_1大于1时具有相同值的任何实例。

我只想捕获列id_4和id_5。这将最终添加到JSON对象中，因此最终的结果是：

{"id_1": "0133", "id_2": "11", "id_3": "kelly", "items": [{"id_4":"AA-1", "id_5":"1"}]}
{"id_1": "2119", "id_2": "22", "id_3": "Wade", "items": [{"id_4":"AA-2", "id_5":"1"}]}
{"id_1": "3903", "id_2": "33", "id_3": "John", "items": [{"id_4":"BB-1", "id_5":"1",{"id_4":"BB-1", "BB-2":"1"}, ,{"id_4":"BB-1", "BB-3":"1"}]}
{"id_1": "5133", "id_2": "44", "id_3": "Emily", "items": [{"id_4":"C-1", "id_5":"1"}]}
{"id_1": "9148", "id_2": "99", "id_3": "Pete", "items": [{"id_4":"BB-34", "id_5":"1",{"id_4":"BB-23", "BB-2":"1"}]}
{"id_1": "2910", "id_2": "111", "id_3": "Mark", "items": [{"id_4":"DD-3", "id_5":"1"}]}

有人知道实现这样的目标的最佳方法吗？任何洞察力都会受到极大的赞赏。

Answer 1

你可以尝试这样的方法：

df.groupby('id_1').nth(1).reset_index()[['id_4','id_5']]

然后，您可以将其转换为JSON。