Computer >> Máy Tính >  >> Lập trình >> Python

Mảng tổng tiền tố trong python sử dụng hàm tích lũy

Cho một mảng và chúng ta phải thực hiện mảng tổng tiền tố bằng cách sử dụng hàm môđun function.itertools.accumulate (iterable [, func]) tất cả các hàm tạo và trả về các trình vòng lặp. Vì vậy, chúng chỉ nên được truy cập bằng các hàm hoặc vòng lặp cắt ngắn luồng. Tạo một trình lặp trả về các tổng tích lũy. Các phần tử có thể là bất kỳ kiểu có thể thêm nào bao gồm Thập phân hoặc Phân số. Nếu đối số hàm tùy chọn được cung cấp, nó phải là một hàm gồm hai đối số và nó sẽ được sử dụng thay vì bổ sung.

Ví dụ

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

Thuật toán

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

Mã mẫu

# 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)

Đầu ra

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]