본문 바로가기

Python(알고리즘,문제풀이)/프로그래머스(코딩기초트레이닝)

코딩기초트레이닝 / 빈 배열에 추가,삭제하기 - 파이썬 (POP)

728x90

📚문제

출처 : 프로그래머스(https://school.programmers.co.kr/learn/courses/30/lessons/181860)

 

📝풀이

def solution(arr, flag):
    x = []
    for idx,k in enumerate(flag):
        if k == True:
            for _ in range(arr[idx]*2):
                x.append(arr[idx])
        else:
            for _ in range(arr[idx]):
                x.pop()
    return x

 

리스트의 가장 마지막의 값을 제거하는 pop()함수를 사용해봤다

pop()함수에서 중요한 점은

 

1. list.pop() => 가장 마지막 값(가장 오른쪽 값) 반환

# Example
my_list = [1,2,3,4,5]
------------------------
my_list.pop() => 5 반환
my_list => [1,2,3,4] 반환

 

+ 추가로 pop()안에 인덱스를 지정하여 빼낼 수도 있다

# Example
my_list = [1,2,3,4]
-----------------------
popped = my_list.pop(1)
popped => 2

pop(1)로 지정하여

my_list의 1번 인덱스인 2 추출 가능

 

2. my_list.remove(), my_list.append() 함수와의 공통점/차이점

공통점 => 리스트를 변경함 

차이점 => 반환값이 있음(가장 오른쪽의 값 or 인덱스로 지정해준 값)

728x90