잘라서 배열로 저장하기

문자열 my_str과
n이 매개변수로 주어질 때,
my_str을 길이 n씩 잘라서 저장한 배열을
return하도록 solution 함수를 완성해주세요.

내 코드

import math

def solution(my_str, n):
    answer = []
    for i in range(0,math.ceil(len(my_str)/n) ):
        if len(my_str) >= n:
            answer.append(my_str[0:n])
            my_str = my_str[n:]
        else:
            answer.append(my_str[0 : len(my_str) % n + 1])
            my_str = my_str[len(my_str) % n + 1:]        
    return answer

print(solution("abcdef123",3))
print(solution("abc1Addfggg4556b",6))

예시 코드

def solution(my_str, n):
    return [my_str[i: i + n] for i in range(0, len(my_str), n)]
    
print(solution("abcdef123",3))
print(solution("abc1Addfggg4556b",6))

Python에서 문자열을 슬라이싱 할 때
인덱스가 문자열의 길이를 초과하더라도 에러가 발생하지 않습니다.
대신 가능한 범위 내에서 슬라이싱이 이루어집니다.