원하는대로

관심분야에 대해 원하는 모든 것을 발행하는 곳

미정 자세히보기

공부 스걱스걱/코딩테스트

[python3]프로그래머스_신규 아이디 추천

ohsoou 2021. 5. 17. 21:03

정규식 찾아보느라 좀 오래 걸렸다.

이 문제를 풀면서 정규식에 대한 이해를 조금이나마 더 하게 되었다.

import re
def solution(new_id):
    answer = ''
    new_id = new_id.lower()
    new_id = re.sub('[^a-z0-9-_.]', '', new_id)
    new_id = re.sub('[.]{1,}', '.', new_id).strip('.')
    if not new_id :
        new_id = "a"
    if len(new_id) >= 16 :
        new_id = new_id[:15].rstrip('.')
    if len(new_id) <= 2:
        while(len(new_id) < 3):
            new_id += new_id[-1]
            
    answer = new_id
    return answer