Python中按复合键分组问题：如何保留CSV所有字段？

我之前也碰到过类似的困扰！用itertools.groupby处理这种每条路线带固定id/alias、同时包含多个点位的CSV时，很容易踩两个坑：忘记先排序，或者不知道怎么正确提取路线的元数据。下面给你一步步拆解解决方案：

问题核心分析 #

itertools.groupby的工作逻辑是把连续相同key的元素归为一组，不是全局分组——如果你的CSV里同一路线的点位不是连续排列的，直接分组会得到零散的小分组。另外，分组后的key只是你指定的分组依据（比如路线id），并不会自动包含alias这类元数据，得从组内的元素里提取。

完整代码示例（读取+分组+输出所有字段） #

假设你的CSV结构是这样的：

id,alias,longitude,latitude
1,路线A,116.397,39.908
1,路线A,116.401,39.907
2,路线B,121.473,31.230
2,路线B,121.475,31.228

用下面的代码可以实现分组并输出所有字段：

import csv
from itertools import groupby

# 读取原始CSV
with open("your_routes.csv", "r", newline="") as in_file:
    reader = csv.DictReader(in_file)
    # 第一步：必须按路线id排序，确保同一路线的点位连续
    sorted_data = sorted(reader, key=lambda row: row["id"])

    # 按路线id分组
    for route_id, group in groupby(sorted_data, key=lambda row: row["id"]):
        # 把分组转换成列表（group是迭代器，遍历一次就失效，转列表方便复用）
        route_items = list(group)
        # 提取该路线的alias（同一组的alias都相同，取第一个元素即可）
        route_alias = route_items[0]["alias"]

        # 输出路线元数据+所有点位信息
        print(f"=== 路线ID: {route_id} | 别名: {route_alias} ===")
        for item in route_items:
            print(f"点位：经度{item['longitude']}, 纬度{item['latitude']}")
        print("\n")

关键细节说明 #

1. 必须先排序：这是groupby的核心要求，如果跳过这一步，同一路线的点位分散在文件中时，会被分成多个独立分组，完全达不到预期效果。
2. 将group转为列表：groupby返回的组是一个迭代器，如果你先遍历了组内的点位，就没法再回头取alias了——转成列表后可以随意访问组内的任意元素。
3. 提取元数据：因为同一路线的所有点位的id和alias都是一致的，所以直接取组内第一个元素的alias即可，不用重复读取。

如果要输出到新CSV文件 #

如果需要把分组后的结果保存为包含所有字段的新CSV，代码更简单：

import csv
from itertools import groupby

with open("your_routes.csv", "r", newline="") as in_file, open("grouped_routes.csv", "w", newline="") as out_file:
    reader = csv.DictReader(in_file)
    writer = csv.DictWriter(out_file, fieldnames=reader.fieldnames)
    writer.writeheader()

    sorted_data = sorted(reader, key=lambda row: row["id"])
    for _, group in groupby(sorted_data, key=lambda row: row["id"]):
        # 直接写入组内的所有记录，每条记录都包含id、alias、经纬度
        for row in group:
            writer.writerow(row)

这样生成的新CSV会把同一路线的点位连续排列，同时保留所有原始字段。

内容的提问来源于stack exchange，提问作者6659081