>  기사  >  백엔드 개발  >  Python에서 데이터를 열별로 그룹화하고 목록 사전으로 변환하는 방법은 무엇입니까?

Python에서 데이터를 열별로 그룹화하고 목록 사전으로 변환하는 방법은 무엇입니까?

DDD
DDD원래의
2024-10-29 13:46:29711검색

How to Group Data by Column and Convert it to a Dictionary of Lists in Python?

GroupBy 결과를 목록 사전으로

Column1, Column2, Column3의 3개 열이 있는 Excel 스프레드시트가 있습니다. 데이터를 추출하여 Column1별로 그룹화하고 다음과 같은 목록 사전을 생성하려고 합니다.

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

이 결과를 얻으려면 다음 코드를 사용할 수 있습니다.

excel = pandas.read_excel(r"e:\test_data.xlsx", sheetname='mySheet', parse_cols="A,C")
grouped_data = excel.groupby("Column1")["Column3"].apply(list).to_dict()

groupby 함수는 데이터를 Column1별로 그룹화하고 groupby 개체를 반환합니다. 그런 다음 Apply 메서드를 사용하여 Column3 데이터에 목록 함수를 적용하여 각 그룹에 대한 값 목록을 만듭니다. 마지막으로 to_dict 메서드는 그룹화된 데이터를 목록 사전으로 변환하는 데 사용됩니다.

결과 사전은 사용자가 지정한 형식을 가지며, 키는 Column1의 값이고 값은 Column1의 목록입니다. 3열에 해당 값이 있습니다.

위 내용은 Python에서 데이터를 열별로 그룹화하고 목록 사전으로 변환하는 방법은 무엇입니까?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!

성명:
본 글의 내용은 네티즌들의 자발적인 기여로 작성되었으며, 저작권은 원저작자에게 있습니다. 본 사이트는 이에 상응하는 법적 책임을 지지 않습니다. 표절이나 침해가 의심되는 콘텐츠를 발견한 경우 admin@php.cn으로 문의하세요.