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

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

<시간/>

주어진 배열과 우리는 접두사 sum 배열을 사용하여 접두사 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]