使用旧列表中的复制和平均项目创建新列表

Question

我有一个由几个子列表组成的主列表，这些子列表又由子列表组成（即：子子列表）这是我的意思的一个最小示例：

a = [[[0,1,2,3], [0,4,5,6]], [[1,1,2,7], [1,4,5,8]], [[2,1,2,9], [2,4,5,10]]]

注意每个子子列表中的0项表示其所在子列表的索引，1和2项在每个子子列表中的两个子子列表中是等价的子列表和第三项不同。

我需要基于该列表创建一个新列表，方法是省略零项，然后从每个子子列表（相等）复制项目1 和2，最后平均第三项。所以新列表看起来像这样：

b = [[1,2,6.33], [4,5,8.]]

在零项消失的地方，1 和 2 项只是被复制，第三项在子子列表中平均：

6.33 = (3+7+9)/3
8. = (6+8+10)/3

我确信这可以使用zip 和np.mean 完成，但我无法完成。顺便说一句，我可以接受不使用这些工具的答案，这只是我对答案可能在哪里的直觉。

Answer 1

我认为这可行：

a = [[[0,1,2,3], [0,4,5,6]], [[1,1,2,7], [1,4,5,8]], [[2,1,2,9], [2,4,5,10]]]
arr = np.asarray(a)

b =  arr[:, :, 1:].mean(0)

这取决于子列表的1 和2 项是相同的，因此它们的平均值不会改变。

>>> b
array([[ 1.        ,  2.        ,  6.33333333],
       [ 4.        ,  5.        ,  8.        ]])