[프래그래머스] 가운데 글자 가져오기(Python)

문제 설명

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

첫 코드

def solution(s):
    if len(s) % 2 == 1:
        i = len(s)-len(s)//2-1
        answer = s[i]
    if len(s) % 2 == 0:
        i = len(s)//2
        answer = s[i-1] + s[i]
    return answer

개선된 코드

셋째 줄의 'len(s) - len(s)//2- 1' 거추장스러운 코드를 'len(s)//2'로 만들 수 있습니다.

그리고 나머지 부분도 깔끔하게 정리해 보았습니다.

def solution(s):
	i = len(s)//2
	if len(s) % 2 == 1:
    	answer = s[i]
    else:
    	answer = s[i-1] + s[i]
	return answer