python - extend

정의

extend 메소드는 리스트에 다른 리스트의 모든 요소를 추가함 

예를 들어, list1.extend(list2)는 list1의 끝에 list2의 모든 요소를 추가함 

 

양식

list1.extend(iterable)

list1: 요소가 추가될 리스트
iterable: 리스트에 추가될 요소를 가진 iterable 객체(예: 리스트, 튜플, 문자열 등)
extend 메소드는 iterable에 있는 모든 요소를 가져와서 list1의 끝에 추가 
이 메소드는 원래의 리스트(list1)를 직접 수정하며, 새로운 리스트를 생성하지 않음

 

예시 코드1

list1 = [1, 2, 3]
list2 = [4, 5, 6]

# list1에 list2의 모든 요소를 추가
list1.extend(list2)

print(list1)



출력값: [1, 2, 3, 4, 5, 6]

 

예시 코드2 

def solution(arr):
    answer = []
    for i in arr:
        answer.extend([i]*int(i))
    return answer

 

위와 같은 코드에 

arr = ['3', '1', '4']를 넣는다면 

1. '3'에 대해 [3, 3, 3]을 만들어서 answer에 추가
2. '1'에 대해 [1]을 만들어서 answer에 추가
3. '4'에 대해 [4, 4, 4, 4]를 만들어서 answer에 추가

answer = ['3', '3', '3', '1', '4', '4', '4', '4']

최종적으로 이렇게 된다