[프로그래머스] Python3 세로 읽기

문제 설명
문자열 my_string과 두 정수 m, c가 주어집니다. my_string을 한 줄에 m 글자씩 가로로 적었을 때 왼쪽부터 세로로 c번째 열에 적힌 글자들을 문자열로 return 하는 solution 함수를 작성해 주세요.

def solution(my_string, m, c):
    answer = []
    strings = []
    start=0
    end=m
    
    if 1 == m == c :
        answer = my_string
    else : 
        for i in range(0, len(my_string), m):
            strings.append(my_string[start:end])
            start = start + m
            end = end + m
        
        for j in range(len(strings)):
            if c <= len(strings[j]):
                answer.append(strings[j][c-1])
        
        answer = ''.join(answer)
        
    return answer