时间戳

首页 > 解决方案 > 如何从共享相同键的元组列表中找到平均值？

python-3.x - 如何从共享相同键的元组列表中找到平均值？

问题描述

我有两个从用户输入创建的列表，我使用以下代码将它们组合成一个元组列表：

daily_hours_list = [4, 2, 1, None, 3, 5]
week_counter_list = [1, 1, 1, 2, 2, 2]

weekly_hours_list = []
for week, time in zip(week_counter_list, daily_hours_list):
    if time != None:
        weekly_hours_list.append((week, t))

这给了我：

weekly_hours_list = [(1, 4),
                     (1, 2),
                     (1, 1),
                     (2, 3),
                     (2, 5)]

然后，我使用此代码将第 1 周的所有时间和第 2 周的所有时间相加：

tup_h = {i:0 for i, v in weekly_hours_list}
for key, value in weekly_hours_list:
    tup_h[key] = tup_h[key]+value
weekly_sum_hours = list(map(tuple, tup_h.items()))

给我：

weekly_sum_hours = [(1, 6),
                    (2, 8)]

这一切都很好，但我如何找到每周的平均小时数，例如：

weekly_average_list = [(1, 2),
                       (2, 4)]

我想我需要扩展 for 循环计算，考虑周值为 1 和 2 的元组计数，但不知道如何实现。我在这里先向您的帮助表示感谢。

标签： python-3.xlisttuplesuser-input

解决方案

我认为有帮助的是首先收集每周的时间。这可以通过字典轻松完成，其中键是周数，值是该周的小时列表。defaultdict内置模块中有一个称为 a 的数据结构collections，它是专门为这样的情况设计的：

from collections import defaultdict
from statistics import mean

daily_hours_list = [4, 2, 1, None, 3, 5]
week_counter_list = [1, 1, 1, 2, 2, 2]

daily_hours_by_week = defaultdict(list)
for week, time in zip(week_counter_list, daily_hours_list):
    if time is not None:
        daily_hours_by_week[week].append(time)

sum_hours_by_week = {w: sum(hours) for w, hours in daily_hours_by_week.items()}
avg_hours_by_week = {w: mean(hours) for w, hours in daily_hours_by_week.items()}

在我们的示例中，这意味着您不必为每个周数使用一个空列表来初始化字典（这是您对 for 的初始总和所做的0）tup_h。相反，如果我们尝试将一个小时附加到字典中尚未出现的一周，它将创建一个空列表放在该键下，然后附加到该键。

一旦我们像这样安排每周的工作时间，就很容易对它们进行其他处理。

实际上，我们可以一次完成最后两行，并创建一个带有每周统计数据元组的字典：

statistics_by_week = {w: sum(hours), mean(hours) for w, hours in daily_hours_by_week.items()}

defaultdict在此处阅读更多详细信息： https ://docs.python.org/3/library/collections.html#collections.defaultdict

推荐阅读