如何计算方差？

Question

我的代码的目的是在列表列表中找到平均值和方差。约束是：如果在“linesort”中有两个或更多列表，其中前两个元素相等，则计算添加列表的第三个元素的平均值。我的问题是包括方差和平均值的计算，并返回一个包含 [a, b, mean,variance] 的列表。非常感谢你。linesort = [[1, 2, 3.00], [1, 2, 5.00], [1, 4, 7.00], [1, 4, 3.00] ,[3, 6, 5.2]]new = []final = []count=0for el in linesort:    new.append(el[:-1])tnew = [tuple(t) for t in new]setnew = set(tnew)setnew = [list(t) for t in setnew]for items in setnew:    inds = [i for i,x in enumerate(new) if x == items]    if len(inds) > 1:        somma = 0        for ind in inds:            print(somma)            somma = linesort[ind][2] + somma        media = somma/len(inds)        items.append(media)        final.append(items)print(final)期望输出：('Output: ', [[1, 2, 4.0,1.0], [1, 4, 5.0,4.0]])至于差异，我想过这行代码，但我无法让它工作。variance = float(sum((linesort[ind][2] - media) ** 2 for linesort[ind][2] in linesort) / len(linesort))

Answer 1

您可以通过首先重新组织字典中的数据来简化代码，将前两个元素的元组作为键，并将相应的值放在列表中。

您可以使用 adefaultdict使这更容易。

然后，我们只需要计算每个列表的均值和方差。

from collections import defaultdict

linesort = [[1, 2, 3.00], [1, 2, 5.00], [1, 4, 7.00], [1, 4, 3.00] ,[3, 6, 5.2]]

# Let's first group the values: 

d = defaultdict(list)

for x, y, val in linesort:

    d[(x, y)].append(val)

# d will be: {(1, 2): [3.0, 5.0], (1, 4): [7.0, 3.0], (3, 6): [5.2]}    

# Now we can build the output list:

out = []

for (x, y), values in d.items():

    n = len(values)

    mean = sum(values)/n

    variance = sum(x**2 for x in values)/n - mean**2

    out.append([x, y, mean, variance])

print(out)

# [[1, 2, 4.0, 1.0], [1, 4, 5.0, 4.0], [3, 6, 5.2, 0.0]]

要回答您的评论：

如果您想省略只有一个值的情况，只需将最后一部分更改为：

for (x, y), values in d.items():

    n = len(values)

    if n > 1:

        mean = sum(values)/n

        variance = sum(x**2 for x in values)/n - mean**2

        out.append([x, y, mean, variance])