按結果分組到列表字典

您有一個包含三列的 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.