Search code examples
pythonarrayssum

Print Sum of Array

import math

import os

import random

import re

import sys


def simpleArraySum(ar_count,ar):
    if ar_count == len(ar):
         print(sum(ar))
    else:
        print("Invalid Input")
    


if __name__ == '__main__':
    fptr = open(os.environ['OUTPUT_PATH'], 'w')

    ar_count = int(input().strip())

    ar = list(map(int,input().rstrip().split()))

    result = simpleArraySum(ar_count,ar)

    fptr.write(str(result) + '\n')

    fptr.close()

I am trying to print sum of values in an array in python. I am trying to solve this on a coding platform but it does not print any value. Why my code is not giving any kind of Output???

Solution

  • Try return instead of printing the outputs.

    def simpleArraySum(ar_count,ar):
    if ar_count == len(ar):
         return sum(ar)
    else:
        return "Invalid Input"