首頁 > 後端開發 > Python教學 > 如何在 Python 中按列對資料進行分組並將其轉換為列表字典?

如何在 Python 中按列對資料進行分組並將其轉換為列表字典?

DDD
發布: 2024-10-29 13:46:29
原創
759 人瀏覽過

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

將結果分組到清單字典

您有一個包含三列的 Excel 電子表格:Column1、Column2 和 Column3。您想要提取數據,按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 中的值Column3.

中對應的值

以上是如何在 Python 中按列對資料進行分組並將其轉換為列表字典?的詳細內容。更多資訊請關注PHP中文網其他相關文章!

來源:php.cn
本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板