python - 如何根据defaultdict中的键累积值

Question

如果我有一个 defaultdict(list)

d =defaultdict(list)
such as 1: 0
        1:0.2
        1: 0.3
        2: 0.2
        2: 0.4
        2: 0.1

......我如何将其转换为类似的 defaultdict(list) 其值是具有相同键的第一个 dict 的值的累积，但从不大于 1

到目前为止我只有以下代码，它不太正确:-(

 d2 = defaultdict(list)
 for k in d.iterkeys()
          v +=d(k)
          d2[k]. append(v)

score 1 · Accepted Answer

原始帖子有点令人困惑：这些示例看起来像一个多图而不是列表的字典。

我假设您有一个带有数字列表的 defaultdict，并且您想创建一个新的 defaultdict，其中包含每个列表中值的累积总和。尝试这个：

d2 = defaultdict(list)
for k, v in d.iteritems():
    sum = 0.0
    v2 = []
    for x in v:
        sum += x
        v2.append(sum)
    d2[k] = v2

请注意，当您添加新值时，新的 defaultdict 不会自动计算累积总和。

如果您安装了 numpy 之类的库，则可以使用numpy.cumsum而不是使用 python 循环来计算累积和。

我不清楚“从不大于 1”到底是什么意思。累积和达到 1 后是否要忽略元素？您是否要忽略大于 1 的单个值，但继续累积较小的值，即使总和超过 1？您是否希望每个输出列表中的元素数量相同，但将元素值限制为 1？

score 0 · Accepted Answer

作为参考，您可以更优雅地执行此操作：

d2 = defaultdict(float)
for k, v in d.iteritems():
    d2[k] += v

2 回答 2