輸入--> a = [[297, 151, 320], [293, 151, 305], [296, 151, 320], [295, 162, 306], [297, 160, 309], [300, 158, 321]我在列表中有一個列表。我需要按列表 a[i][2] 中的第三個元素分組，公差 + 或 - 5output_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]]]有人能幫忙嗎 ？
2 回答
aluckdog
TA貢獻1847條經(jīng)驗 獲得超7個贊
對不起，我沒有得到你想要的。因為每個列表的 ur 值不一致。但是，你的意思是像這個嗎？
a = [[297, 151, 320], [293, 151, 305], [296, 151, 320], [295, 162, 306], [297, 160, 309], [300, 158, 321]]
b = []
i = -1
for 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]]
DIEA
TA貢獻1820條經(jīng)驗 獲得超2個贊
一種方法是創(chuàng)建一個生成器函數(shù)來進行容差分組。此函數(shù)假定列表已排序，因此您需要傳入一個排序列表，或?qū)ζ溥M行修改以在函數(shù)中進行排序。我相信有人會找到一種方法來做到這一點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
然后在您的排序列表上調(diào)用此函數(shù)（按第 3 項）：
from operator import itemgetter
a = [[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 mean
averages = [[mean(x) for x in zip(*group)] for group in grouped]
print(averages)
平均值：
[[295, 157.66666666666666, 306.6666666666667], [297.6666666666667, 153.33333333333334, 320.3333333333333]]
添加回答
舉報
0/150
提交
取消