Python：对于列表列表，获取每个位置的平均值

Question

我有一个列表：

list_of_lists = []
list_1 = [-1, 0.67, 0.23, 0.11]
list_2 = [-1]
list_3 = [0.54, 0.24, -1]
list_4 = [0.2, 0.85, 0.8, 0.1, 0.9]
list_of_lists.append(list_1)
list_of_lists.append(list_2)
list_of_lists.append(list_3)
list_of_lists.append(list_4)

这个职位很有意义。我想返回一个列表，其中包含每个位置的平均值，不包括-1。也就是说，我想要：

[(0.54+0.2)/2, (0.67+0.24+0.85)/3, (0.23+0.8)/2, (0.11+0.1)/2, 0.9/1]

实际上是：

[0.37, 0.5866666666666667, 0.515, 0.10500000000000001, 0.9]

我怎么能用Python的方式来做呢？你知道吗

编辑：

我使用的是Python2.7，我不是在寻找每个列表的平均值，而是在寻找“位置0处的所有列表元素不包括-1”的平均值，以及“位置1处的所有列表元素不包括-1”的平均值，等等

原因是：

[(0.54+0.2)/2, (0.67+0.24+0.85)/3, (0.23+0.8)/2, (0.11+0.1)/2, 0.9/1]

位置0的值是-1、-1、0.54和0.2，我想排除-1；位置1有0.67、0.24和0.85；位置3有0.23、-1和0.8，依此类推

Answer 1

这是一个基于矢量化numpy的解决方案。你知道吗

import numpy as np

a = [[-1, 0.67, 0.23, 0.11],
     [-1],
     [0.54, 0.24, -1],
     [0.2, 0.85, 0.8, 0.1, 0.9]]

# first create non-jagged numpy array
b = -np.ones([len(a), max(map(len, a))])

for i, j in enumerate(a):
    b[i][0:len(j)] = j

# count negatives per column (for use later)
neg_count = [np.sum(b[:, i]==-1) for i in range(b.shape[1])]

# set negatives to 0
b[b==-1] = 0

# calculate means
means = [np.sum(b[:, i])/(b.shape[0]-neg_count[i]) \
         if (b.shape[0]-neg_count[i]) != 0 else 0 \
         for i in range(b.shape[1])]

# [0.37,
#  0.58666666666666667,
#  0.51500000000000001,
#  0.10500000000000001,
#  0.90000000000000002]

Answer 2

您可以使用熊猫模块过程代码希望这样：

import numpy as np
import pandas as pd

list_1 = [-1, 0.67, 0.23, 0.11,np.nan]
list_2 = [-1,np.nan,np.nan,np.nan,np.nan]
list_3 = [0.54, 0.24, -1,np.nan,np.nan]
list_4 = [0.2, 0.85, 0.8, 0.1, 0.9]
df=pd.DataFrame({"list_1":list_1,"list_2":list_2,"list_3":list_3,"list_4":list_4})
df=df.replace(-1,np.nan)
print(list(df.mean(axis=1)))

Answer 3

没有第三方库的解决方案：

from itertools import zip_longest
from statistics import mean

def f(lst):
    return [mean(x for x in t if x != -1) for t in zip_longest(*lst, fillvalue=-1)]

>>> f(list_of_lists)
[0.37, 0.5866666666666667, 0.515, 0.10500000000000001, 0.9]

它使用^{}，fillvalue设置为-1来“转置”列表，并将缺少的值设置为-1（将在下一步被忽略）。然后，使用生成器表达式和^{}过滤出-1并得到平均值。你知道吗