我是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)
我知道这是错的,但是有办法做到这一点吗?我确信我忽略了一些非常简单的事情......但我想不出另一种方法可以做到这一点。
答案 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))