[프로그래머스][1단계][python] 가운데 글자 가져오기 문제

문제 설명
단어 s의 가운데 글자를 반환하는 함수, solution을 만들어 보세요. 
단어의 길이가 짝수라면 가운데 두글자를 반환하면 됩니다.

제한 사항
s는 길이가 1 이상, 100이하인 스트링입니다.

입출력 예
입력 : abcde / 출력 : c
입력 : qwer / 출력 : we

 

def solution(s):
    # 중간값 구하기 위해 2나눈 것의 몫구함
    center = len(s)//2

    # 문자열 길이가 짝수일때 가운데 문자 2개 가져옴
    if(len(s)%2 == 0):
        answer = s[center-1:center+1]
    # 문자열 길이가 홀수일때 가운데 문자 1개 가져옴
    else:
        answer = s[center]

    return answer