将嵌套列表的按元素的均值插入同一列表

时间:2019-03-28 11:08:21

标签: python list nested-lists

假设有一个嵌套的float列表

L = [[a,b,c],[e,f,g],[h,i,j]]

我可以定义什么样的函数来遍历列表一次并将每个连续列表的元素均值插入同一列表?即我想得到

L1 = [[a,b,c],[(a+e)/2,(b+f)/2,(c+g)/2],[e,f,g],[(e+h)/2,(f+i)/2,(g+j)/2],[h,i,j]]

我知道获取两个列表的元素均值的函数:

from operator import add
new_list = list(map(add,list1,list2))
J = [j/2 for j in new_list]

然而,将平均值列表重新插入同一列表中,同时又要在旧列表中保持适当的索引迭代是很困难的。

1 个答案:

答案 0 :(得分:1)

有两种情况:

  1. 您不在乎结果列表是否与列表相同:
new_list = []
for i in range(len(L)-1):
    new_list.append(L[i])
    new_list.append(list(map(lambda x: sum(x)/len(x), zip(L[i],L[i+1]))))
new_list.append(L[-1])
  1. 您希望更改就地完成:
i=0
while i < len(L)-1:
    new_elem = list(map(lambda x: sum(x)/len(x), zip(L[i],L[i+1])))
    L.insert(i+1, new_elem)
    i += 2

编辑:如果您使用的是python 3.4或更高版本,则可以使用lambda x: sum(x)/len(x)(来自软件包mean(x))来代替statistics