如何在 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中文网其他相关文章！