如何使这个简单的列表理解?

时间:2010-03-28 10:19:36

标签: python list-comprehension

我是python的新手,我正在努力更好地了解列表推导 我甚至不确定列表理解是否是我正在寻找的词,因为我没有生成列表。但我正在做类似的事情。

这就是我想要做的事情:

我有一个数字列表,其长度可以被三整除。

所以说我有nums = [1, 2, 3, 4, 5, 6] 我想迭代列表并得到每组三位数的总和。 目前我这样做:

for i in range(0, len(nums), 3):
    nsum = a + b + c for a, b, c in nums[i:i+3]
    print(nsum)

我知道这是错的,但是有办法做到这一点吗?我确信我忽略了一些非常简单的事情......但我想不出另一种方法可以做到这一点。

3 个答案:

答案 0 :(得分:6)

请参阅sum(iterable[, start])内置,并在切片上使用它。

  

从左到右汇总开始和迭代的项目并返回总数。 start默认为0. iterable的项通常是数字,不允许是字符串。

>>> nums
[1, 2, 3, 4, 5, 6]
>>> [sum(nums[i:i+3]) for i in  range(0, len(nums),3)]
[6, 15]
>>> 

答案 1 :(得分:4)

import itertools

nums = [1, 2, 3, 4, 5, 6]

print [a + b + c for a, b, c in itertools.izip(*[iter(nums)] * 3)]

答案 2 :(得分:1)

nums = [1, 2, 3, 4, 5, 6]
map(sum, itertools.izip(*[iter(nums)]*3))