Capitalize! Python Hackerrank solution

You are asked to ensure that the first and last names of people begin with a capital letter in their passports. For example, alison heck should be capitalised correctly as Alison Heck.

Given a full name, your task is to capitalize the name appropriately.

alison heck => Alison Heck

Input Format

A single line of input containing the full name,S.

Constraints

0<len(S)<1000

The string consists of alphanumeric characters and spaces.

Note: in a word only the first character is capitalized. Example 12abc when capitalized remains 12abc.

Output Format

Print the capitalized string,S.

Sample Input

chris alan

Sample Output

Chris Alan

Solution

def solve(s):
    s = str(s)
    for i in s[:].split():
        s = s.replace(i,i.capitalize())
    return s
if __name__ == '__main__':
    fptr = open(os.environ['OUTPUT_PATH'], 'w')

    s = input()

    result = solve(s)

    fptr.write(result + '\n')

    fptr.close()