Computer >> 컴퓨터 >  >> 프로그램 작성 >> Python

누적 함수를 사용하여 파이썬에서 합계 배열 접두사

<시간/>

배열이 주어지고 function.itertools.accumulate(iterable[, func]) 모듈 함수를 사용하여 접두사 sum 배열을 수행해야 합니다. 모든 함수는 반복자를 생성하고 반환합니다. 따라서 스트림을 자르는 함수나 루프에서만 액세스해야 합니다. 누적 합계를 반환하는 반복자를 만듭니다. 요소는 Decimal 또는 Fraction을 포함하여 추가 가능한 모든 유형일 수 있습니다. 선택적 함수 인수가 제공되는 경우 두 인수의 함수여야 하며 덧셈 대신 사용됩니다.

Input:
Data = [1, 0, 2, 3, 5]
>>> list(accumulate(data))     # running summation
Output:
[1, 1, 3, 6, 11]

알고리즘

Step 1: Create list.
Step 2: Use list(accumulate( ))) function, its return running total. 
Step 3: Display total.

예시 코드

# Python program to print prefix
#sum array using accumulate function 
from itertools import accumulate 
  
def summation(A): 
   print ("The List after Summation ::>", list(accumulate(A))) 
  
# Driver program 
if __name__ == "__main__": 
   A=list()
n=int(input("Enter the size of the First List ::"))
print("Enter the Element of First  List ::")
for i in range(int(n)):
   k=int(input(""))
   A.append(k)
summation(A) 

출력

Enter the size of the First List ::5
Enter the Element of First List ::
1
2
3
4
5
The List after Summation ::> [1, 3, 6, 10, 15]