繁体   English   中英

如何从元组列表中对元组索引 n 处的所有值求和?

[英]How to cumsum all values at tuple index n, from a list of tuples?

在下面的代码中,我想在 for 循环中添加 list=[(0,3),(2,6),(1,10)] 的第二个参数。 第一次迭代应该是 3+6=9,第二次迭代应该添加上一次迭代的输出,即 9 到 10---> 9+10=19,我想要最终输出 S=[9,19]。 我不知道该怎么做,我应该在我的代码中添加另一个循环吗?

T=[(0,3),(2,6),(1,10)]
S=[]
for i in range(len(T)):
    b=T[0][i]+T[0][i+1]
    S.append(b)
  • 使用zip组合来自tuples的值,具有相同的索引。
  • 使用赋值表达式(从蟒3.8)中,在列表理解,以sum在第二值tupleT[1]T
T = [(0,3),(2,6),(1,10)]
T = list(zip(*T))

print(T)
[out]:
[(0, 2, 1), (3, 6, 10)]

# use an assignment expression to sum T[1]
total = T[1][0]  # 3
S = [total := total + v for v in T[1][1:]]

print(S)
[out]:
[9, 19]

使用itertools.accumulate

spam = [(0,3),(2,6),(1,10)]

from itertools import accumulate
print(list(accumulate(item[-1] for item in spam))[1:])

输出

[9, 19]

这应该可以帮助你:

T=[(0,3),(2,6),(1,10)]

lst = [T[i][1] for i in range(len(T))]

final_lst = []

for x in range(2,len(lst)+1):
    final_lst.append(sum(lst[:x]))

print(final_lst)

输出:

[9, 19]

如果你更喜欢list comprehension ,那么使用这一行而不是最后一个for loop

[final_lst.append(sum(lst[:x])) for x in range(2,len(lst)+1)]

输出:

[9, 19]

只需修改您的代码如下

T=[(0,3),(2,6),(1,10)]
S=[]
b = T[0][1]
for i in range(1,len(T)):
    b+=T[i][1]
    S.append(b)

这是本机递归的解决方案

import operator
mylist =[(0,3),(2,6),(1,10)]

def accumulate(L, i, op):
  def iter(result, rest, out):
    if rest == []:
      return out
    else:
      r = op(result, rest[0][i-1])
      return iter(r, rest[1:], out + [r])
  return iter(L[0][i-1], L[1:], [])

print(accumulate(mylist, 2, operator.add))
print(accumulate(mylist, 1, operator.add))
print(accumulate(mylist, 2, operator.mul))

# ==>
# [9, 19]
# [2, 3]
# [18, 180]

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM