Pandas GroupBy 결과를 목록 사전으로 변환하는 방법은 무엇입니까?

GroupBy 결과를 목록 사전으로 변환

Excel 시트에서 데이터를 추출하고 그룹화하려면 groupby를 활용하는 것이 일반적입니다. 팬더 라이브러리의 기능. 그러나 원하는 출력이 키가 그룹 값에 해당하고 값이 해당 요소의 목록인 사전인 경우 기존 접근 방식으로는 충분하지 않을 수 있습니다.

이 특정 요구 사항을 달성하기 위한 한 가지 솔루션은 적용 및 to_dict 함수. 적용 함수를 사용하면 각 그룹에 함수를 적용할 수 있습니다. 이 경우 Column3을 목록으로 변환합니다. 그런 다음 to_dict 함수는 결과를 사전으로 변환합니다. 다음 코드는 이 접근 방식을 보여줍니다.

<code class="python">import pandas as pd

excel = pd.read_excel(r"e:\test_data.xlsx", sheetname='mySheet', parse_cols'A,C')
result = excel.groupby('Column1')['Column3'].apply(list).to_dict()</code>

이 코드는 원하는 출력을 생성합니다.

{0: [1],
1: [2, 3, 5],
2: [1, 2],
3: [4, 5],
4: [1],
5: [1, 2, 3]}

또는 더 간단한 솔루션은 사전 이해를 사용하는 것입니다.

<code class="python">result = {k: list(v) for k, v in excel.groupby('Column1')['Column3']}</code>

이 접근 방식은 원하는 것과 동일한 출력을 생성하여 간결하고 효율적인 구현을 제공합니다.

위 내용은 Pandas GroupBy 결과를 목록 사전으로 변환하는 방법은 무엇입니까?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!