Itertools分组问题

Question

我正在努力理解itertools.groupby是如何工作的。我有一个excel电子表格，它的交付日期在第一列，然后是目标Lat在第二列，目标Lon在第三列。之前，我能够帮助将相同的日期分组到容纳它们的较大数组的子数组中。下面是执行此操作的代码。

with xlrd.open_workbook(file_location) as workbook:
    sheet = workbook.sheet_by_index(0)

    Dates = (sheet.cell_value(i,0) for i in range(sheet.nrows))
    Day = [list(group) for key, group in itertools.groupby(Dates)]

现在，我需要更进一步，将Lat分组到一个数组中，将Lon分组到另一个数组中，但按日期对它们进行分组。我尝试将上面列出的代码与类似这样的代码组合在一起，但我不知道如何将Lat和Lon变量合并到itertools groupby函数中。

with xlrd.open_workbook(file_location) as workbook:
    sheet = workbook.sheet_by_index(0)

    for i in range(sheet.nrows):
        Lat = (sheet.cell_value(i,2) for i in range(sheet.nrows))
        DeliveryX = [list(group) for key, group in itertools.groupby(Dates)]
        Lon = (sheet.cell_value(i,3) for i in range(sheet.nrows))
        DeliveryY = [list(group) for key, group in itertools.groupby(Dates)] 

Answer 1

groupby接受key参数来决定如何对给定列表中的值进行分组。实际上，您将创建一个可迭代的元组，如下所示：

t = ((lat, lon, date), (lat, lon, date), ...)

您可以通过调用itertools.izip(Lat, Lon, Dates)来实现此目的。然后，要通过比较第三个字段将此列表分成组，并提取第一个和第二个字段，您可以这样写

lats = [[i[0] for i in g] for k, g in groupby(t, key=lambda x: x[2])]
lats = [[i[1] for i in g] for k, g in groupby(t, key=lambda x: x[2])]