[프로그래머스] 잘라서 배열로 저장하기 with 파이썬

my_str = "abc1Addfggg4556b" 

n = 6

답 : ["abc1Ad", "dfggg4", "556b"]

 

1. range(시작, 끝, 간격) : n개씩 자르기 == 간격이 n인 상황

def solution(my_str, n):
    result = []
    for i in range(0, len(my_str), n):
        if len(my_str[i:i + n]) >= 6:
            result.append(my_str[i:i + n])
        else:
            result.append(my_str[i:len(my_str)])

    return result

 

1-1. slicing의 경우, 범위를 초과해도 에러가 나지 않기 때문에 1번처럼 len()으로 조건을 처리해주지 않아도 되긴 한다.

def solution2(my_str, n):
    return [my_str[i:i + n] for i in range(0, len(my_str), n)]

 

 

2. while list/string : list/string이 있을 때까지

def solution3(my_str, n):
    result = []
    while my_str:
        if len(my_str) >= n:
            result.append(my_str[:n])
            my_str = my_str[n:]
        else:
            result.append(my_str)
            my_str = ''
    return result