📅  最后修改于: 2023-12-03 15:18:56.550000             🧑  作者: Mango
Python函数 meanGroups(a) 是一个用来计算列表a中每组元素的平均值的函数。它能够将a中具有相同平均值的元素分为一组,并返回一个由这些组成的列表。
a = [[3,3,4,2], [4,4], [4,0,3,3], [2,3], [3,3,3]]
meanGroups(a)
[[0, 2], [1], [3, 4]]
以下是 meanGroups(a) 函数的代码实现:
def meanGroups(a):
avg_dict = {}
for i in range(len(a)):
avg = sum(a[i])/len(a[i])
if avg not in avg_dict:
avg_dict[avg] = [i]
else:
avg_dict[avg].append(i)
return [list(map(lambda x: a[x], avg_dict[key])) for key in avg_dict]
以上代码用到了一个字典avg_dict,它的key是平均值,value是包含了该平均值的元素序号的列表。我们可以循环这个字典,并将所有元素序号都加入一个新的列表中;最后,返回这个新列表即可。