A screening task to be submitted as part of the QOSF Mentorship application.
The solution to the problem can be found in the Juypter notebook titled QOSF Task-1.pynb
here
A discussion of the problem statement and the results is illustrated the PDF titled QOSF Task 1.pdf