按结果分组到列表字典

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