Capitalize in Python - HackerRank Solution

Problem Statement :

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.

alison heck => Alison Heck

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

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.

Solution:

def solve(s):
    result=[]
    name=re.split(r'(\s+)',s)

    for i in range(len(name)):
       result.append(str(name[i]).capitalize())
       
    
    return ''.join(str(ele) for ele in result)
    

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

    s = input()

    result = solve(s)

    fptr.write(result + '\n')

    fptr.close()