Question

Write a python function called "get_sma", that takes two input arguments, similar to the example provided. First argument is the original prices of type list, and the 2nd input argument being the number of days we want to calculate SMA, like a 8 day sma or 35 day sma. This function should return a new list of sma price, based on the input day range, and should be able to handle any SMA calculation, like: sma200 for 200 days moving average.

Example code:

def sma(prices, nday):
sma_data = [0] * len(prices)
for i in range(nday, len(prices)):
sma_data[i] = sum(prices[i-nday:i]) / nday
return sma_data

Answer 1

# Program in Python

import sys # for sys.exit()
# required function which takes a list of prices
# and number of days in SMA and returns a SMA list
def get_sma(prices,nday):
# list to store the calculated SMA
list = []
# iterate over the price list
for i in range(0,n-nday+1):
# local temporary variable to store the intermediate results
temp = 0;
# iterate to find the SMA
for j in range(i,i + nday):
temp += prices[j]
# append the SMA value to the list
list.append(float(temp/nday))
# return the list
return list

# price_list list
price_list = []

# get the number of elements in the input list
n = int(input("Enter number of elements : "))

print ("\nEnter elements in the price list: ")
# store the input as list by iterating within range
for i in range(0, n):
ele = float(input())
# add the element to the list
price_list.append(ele)

# get the number of days in SMA
NoOfDay = int(input("Enter number of SMA Days : "))

# check if days of SMA is greater than size of input list
if n < NoOfDay:
print ("Days of SMA is greater than size of input list!")
# exit from the program
sys.exit(0)
# check if days of SMA is equal to 0
elif NoOfDay == 0:
print ("Invalid Days of SMA!")
# exit from the program
sys.exit(0)
  
# call the get_sma function to get the SMA list
a = get_sma(price_list,NoOfDay)
# print the SMA list
print("Required SMA list is " , a)

# Output

Enter number of elements : 5
Enter elements in the price list: 43.5
20.5
32.25
54
12
Enter number of SMA Days : 4

Required SMA list is [37.5625, 29.6875]

# Screenshot of Code, Input & Output

(Thank You!)