내가 쓰려고 만든 pandas groupby

API를 개발할 때, DB에서 가져온 데이터를 전처리해서
프론트가 사용하기 좋은 형태의 Response를 만들 필요가 있다.

  • 그 때 사용하면 좋은 것이 바로 pandas
result_df = (pd.DataFrame(ui_item_json)
	.groupby(['그룹으로 묶을 기준이 되는 데이터 필드들'])
	.names # 집게하고 싶은 데이터 필드
	.agg(set)
	.reset_index()
    .to_dict("records"))
    
result_df = sorted(result_df, key=lambda d: d['정렬 기준이 될 데이터 필드'])

# group by 이후 특정 필드로 다시 묶고 싶을 경우
res = {}
 
for item in result_df:
	res.setdefault(item['그룹화 기준이 될 데이터 필드'], []).append(item)

return Response(res, status=status.HTTP_200_OK)

좋은 웹페이지 즐겨찾기