Question

这可能很简单，我忽视了一些事情......

我有一长串整数，在这种情况下代表网站的每日访问者。我想要一个新的每周访问者列表。因此，我需要从原始列表中获取七个组，将它们相加，然后将它们添加到新列表中。

我的解决方案似乎相当暴力，不优雅：

numweeks = len(daily) / 7
weekly = []
for x in range(numweeks):
    y = x*7
    weekly.append(sum(visitors[y:y+7]))

是否有更高效或更pythonic的方式来做这件事？

Answer 1

weekly = [ sum(visitors[x:x+7]) for x in range(0, len(daily), 7)]

或稍微不那么密集：

weekly = []
for x in range(0, len(daily), 7):
     weekly.append( sum(visitors[x:x+7]) )

或者，使用numpy模块。

by_week = numpy.reshape(visitors, (7, -1))
weekly = numpy.sum( by_week, axis = 1)

请注意，这要求访问者中的元素数量是7的倍数。它还要求您安装numpy。但是，它可能比其他方法更有效。

或者对于itertools代码奖励：

def grouper(n, iterable, fillvalue=None):
    "grouper(3, 'ABCDEFG', 'x') --> ABC DEF Gxx"
    args = [iter(iterable)] * n
    return itertools.izip_longest(fillvalue=fillvalue, *args)

weekly = map(sum, grouper(7, visitors, 0))

Answer 2

>>> daily = [1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20]
>>> print [sum(daily[x:x+7]) for x in range(0, len(daily), 7)]
[28, 77, 105]

我不确定这是不是“pythonic”，但我真的很喜欢这个python的单行内容。

血腥细节：Comprehensions

Answer 3

使用itertools.islice：

weekly = [sum(list(itertools.islice(daily, i, i+7)))
          for i in range(0, len(daily), 7)]

修改

或者，使用math.fsum：

weekly = [math.fsum(itertools.islice(daily, i, i+7)) for i in range(0, len(daily), 7)]

在python中查找列表的子集总和

3 个答案: