Python：如何在具有容差的列表中对相似列表进行分组并取平均值？

Question

输入--> a = [[297, 151, 320], [293, 151, 305], [296, 151, 320], [295, 162, 306], [297, 160, 309], [300, 158, 321]

我在列表中有一个列表。我需要按列表 a[i][2] 中的第三个元素分组，公差 + 或 - 5

output_1--> a = [[[297, 151, 320], [293, 151, 318], [296, 151, 321]], [[295, 162, 306], [297, 160, 309], [300、158、305]]

稍后我需要取每个子组的平均值，例如 [297, 151, 320], [293, 151, 318], [296, 151, 321] = [(297+293+296)/3,(151+ 151+151)/3,(320+318+321)/3] 同样用于下一组

最终输出

final_output--> a=[[[295,151,320]],[[297,160,307]]]

有人能帮忙吗 ？

Answer 1

对不起，我没有得到你想要的。因为每个列表的 ur 值不一致。但是，你的意思是像这个吗？a = [[297, 151, 320], [293, 151, 305], [296, 151, 320], [295, 162, 306], [297, 160, 309], [300, 158, 321]]b = []i = -1for x in range(len(a)):    if (x%3) == 0:        b.append([])        i += 1    ave = int(sum(a[x])/len(a[x]))    b[i].append(ave)print(b)输出： [[256, 249, 255], [254, 255, 259]]

Answer 2

一种方法是创建一个生成器函数来进行容差分组。此函数假定列表已排序，因此您需要传入一个排序列表，或对其进行修改以在函数中进行排序。我相信有人会找到一种方法来做到这一点itertools.groupby。def groupby_tolerance(lst, tolerance):    result = [lst[0]]    for prev, curr in zip(lst, lst[1:]):        if curr[2] - prev[2] > tolerance:            yield result            result = []        result.append(curr)    yield result然后在您的排序列表上调用此函数（按第 3 项）：from operator import itemgettera = [[297, 151, 320], [293, 151, 305], [296, 151, 320], [295, 162, 306], [297, 160, 309], [300, 158, 321]]grouped = groupby_tolerance(sorted(a, key=itemgetter(2)), 5)这给出了分组：[[[293, 151, 305], [295, 162, 306], [297, 160, 309]], [[297, 151, 320], [296, 151, 320], [300, 158, 321]]]然后你可以压缩相应的元素并计算平均值：from statistics import meanaverages = [[mean(x) for x in zip(*group)] for group in grouped]print(averages)平均值：[[295, 157.66666666666666, 306.6666666666667], [297.6666666666667, 153.33333333333334, 320.3333333333333]]