리스트 컴프리헨션 [리스트]

 이번 글은 문제 풀이의 내용은 아니고, 필자가 헷갈리고 평소에 잘 활용하지 못했던 개념적 내용을 담았다. 그냥 필요할 때마다 개인적으로 들여다보고자 개념을 정리한 것! (말 그대로 내가 보려고 만든 개념 정리 노트..) 앞으로도 코딩테스트나 머신러닝 개념 외에도 python의 문법, 활용 기법들을 자주 정리하려 한다. 

 

📌 리스트 컴프리헨션(list comprehension)

 for 문과 if 문을 리스트 내부에 작성해 코드를 보다 직관적으로 만들고 실행속도를 끌어올려주는 고마운 녀석이다. 여기서 밑의 코드들을 제외하고 counts = [list1.count(i) for i in list1] 라인을 한 번 들여다보자. 

 

✳️ 예시

---

counts =  [ list1 리스트 내부의 i를 돌려보았을 때, i 의 개수들로 이루어진 새로운 리스트]

1) for i in list1

-> list1의 원소들을 하나씩 i로 돌면서 보기.

2) list.count(i)

-> i의 개수를 list1에서 돌면서 count하기.

'''예시) <프로그래머스 코딩 테스트 기초 문제, 181916번,  주사위 게임3> '''

def solution(a, b, c, d):
    
    list1 = [a, b, c, d]
    counts = [list1.count(i) for i in list1]   # list comprehension 기법. 
    
    """ if max(counts) == 4:
        return a * 1111
    elif max(counts) == 3:
        p = list1[counts.index(3)]
        q = list1[counts.index(1)]
        return (10 * p + q) ** 2
    elif max(counts) == 2:
        if min(counts) == 2:
            return ((a + b) * abs(a - b)) if a == c else ((a + c) * abs(a - c)) 
        else:
            p = list1[counts.index(2)]
            return (a * b * c * d) / p ** 2
    else:
        return min(list1)
    """

 

 

✳️ 리스트 내부에 조건 추가하기

---

 리스트 내부에 if 조건문을 사용해 특정값을 걸러내는 filtering 작업이 가능하다. 여기서 조심할 건 and와 or 연산자의 사용, if 조건문의 위치 및 사용 방법이다. 아래는 그 예시로, 문제 조건에 따라 적절하게 활용하면 될 듯하다. 자주 활용되는 방식이니 손에 익을 때까지 여러 문제들을 풀어보면서 익숙해져야겠다. 

 

ex) 리스트명 = [표현식 for 변수 in 반복 가능한 대상 if 조건문]

[ ( 변수를 활용한 값 ) for ( 사용할 변수 이름 ) in ( 순회할 수 있는 값 ) if ( 조건문 ) ]

 

 

#️⃣내부 조건문이 하나일 때.

# 1. if 문이 한 개!
arr = [n for n in range(1, 11) if n % 2 == 0]
print(arr)

# [2, 4, 6, 8, 10]

 

 

#️⃣내부 조건문이 두 개일 때.

# 2. if 문이 두 개!

arr2 = [n for n in range(1, 31) if n % 2 == 0 if n % 3 == 0] 

arr3 = [n for n in range(1, 31) if n % 2 ==0 and n % 3 == 0]
# 같은 조건의 코드.
# 단, 두 개의 if 문 사이에 and 연산자를 사용하는 건 불가능!

 

#️⃣표현식 + 조건문 + for 문의 형태일 때.

# 3. 조건문이 먼저 나올 때.
counts = [n if (n < 15 and n > 9) else 0 for n in range(1, 30)] # else 문은 반드시 써주어야 함.
print(counts)

# [0, 0, 0, 0, 0, 0, 0, 0, 0, 10, 11, 12, 13, 14, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]

# n < 15보다 n > 9인 경우 n 그대로, else : 0으로

 

#️⃣중첩 리스트 컴프리헨션

list1 = [[i for i in range(2)] for j in range(3)]
print(list1)

# [[0, 1], [0, 1], [0, 1]]

이차원 리스트의 생성 과정은 아래와 같다. 

 

j = 0, 1, 2  / i = 0, 1

1) j = 0, [i for i in range(2) -> 0, 1]

- [[0, 1]]

 

2) j = 1, [i for i in range(2) -> 0, 1]

- [[0, 1], [0, 1]]

 

3) j = 2, [i for i in range(2) -> 0, 1]

- [[0, 1], [0, 1], [0, 1]]

 

 

비교적 간단한 예시를 들고 와 개념을 정리해봤는데, 복잡한 컴프리헨션이 요구되는 문제일 경우 많이 까다롭다;; 그리고 컴프리헨션은 비단 list에서만 적용되지 않는다. 딕셔너리와 set에서도 적용이 가능한데, 그 내용은 추가로 다뤄보려 한다. 일단 해당 개념을 자꾸 적용해보면서 익숙해질 필요가 있으니까.. ㅎㅎ