[Python] List Comprension

🌼 파이썬의 꽃, 리스트 컴프리헨션(List Comprehension) 🌼

---

리스트 컴프리헨션이란?

💡 리스트 컴프리헨션은 리스트를 보다 직관적으로 생성하는 방법. 대괄호([]) 안에 for문과 if문을 사용하여 리스트를 간결하게 만들 수 있다.

즉, 여러 줄의 코드를 한 줄로 줄여주며, 가독성을 높여줌!

🤔 리스트 컴프리헨션을 사용한 경우 vs 사용하지 않은 경우

---

1️⃣리스트 컴프리헨션을 사용하지 않고, 0~4까지의 정수를 생성하는 코드:

li = []
for i in range(5):
    li.append(i)

# 출력: [0, 1, 2, 3, 4]

2️⃣리스트 컴프리헨션을 사용한 코드:

li = [i for i in range(5)]

# 출력: [0, 1, 2, 3, 4]

훨씬 간결하고 직관적인걸 볼수있다.

---

[i * 2 for i in range(5)]
# 출력: [0, 2, 4, 6, 8]

응용하면 다양한 형태로 활용할 수 있습니다

import sys
a = [sys.stdin.readline().strip() for _ in range(10)]

10개의 여러 줄 입력을 받는 경우

---

🤔 리스트 컴프리헨션에서 조건문 사용하기

---

💡 리스트 컴프리헨션에서는 for문뿐만 아니라 조건문도 사용이 가능하다.

조건문 사용예시

1️⃣ 짝수를 리스트의 저장할 경우 :

[i for i in range(5) if i % 2 == 0]
# 출력: [0, 2, 4]

2️⃣ 여러 개의 `if` 문을 사용하는 경우:

[i for i in range(5) if i % 2 == 0 if i % 4 == 0]
# 출력: [0, 4]

---

---

if-else 문 사용하기

💡 if-else 문을 사용할 경우 반드시 왼쪽에 배치해야함!!

[i if i % 2 == 0 else 'odd' for i in range(5)]
# 출력: [0, 'odd', 2, 'odd', 4]

---

리스트 컴프리헨션의 장점✅

---

1️⃣ 속도 향상: 일반적인 for 루프보다 빠름 (파이썬 내부 최적화 덕분)
2️⃣ 코드 간결화: 한 줄로 리스트를 생성 가능
3️⃣ 메모리 절약: map()이나 제너레이터 표현식을 활용하면 추가 메모리 사용을 줄일 수 있음

---

리스트 컴프리헨션의 단점 ❌

---

1️⃣ 메모리 사용 증가
- 리스트 컴프리헨션은 모든 요소를 한 번에 메모리에 저장하므로, 대량의 데이터를 처리할 때 메모리 부담이 커질 수 있음.
- 해결책: 제너레이터 표현식을 사용하면 메모리를 절약할 수 있음.

    squares = (x**2 for x in range(10**6))  # 리스트 대신 제너레이터 사용

2️⃣ **불필요한 연산 수행** - 조건문이 많거나 연산량이 많으면 불필요한 계산이 증가할 수 있음. - 해결책: `map()`과 `filter()` 또는 제너레이터 활용.

    squares = (x**2 for x in range(10**6) if x % 2 == 0)

3️⃣ **가독성이 떨어질 수 있음** - 너무 복잡한 리스트 컴프리헨션은 `for` 루프보다 이해하기 어려울 수 있음. - 중첩된 경우 특히 주의해야 함:

    matrix = [[x * y for x in range(5)] for y in range(5)]  # 5x5 행렬 생성

→ 이런 경우에는 **일반적인 `for` 루프가 더 읽기 쉬울 수도 있음**

🔥 정리

✅ 작은 리스트 → 리스트 컴프리헨션이 유리함

✅ 큰 리스트 → 제너레이터 표현식(()) 또는 map(), filter() 고려

✅ 복잡한 조건이나 중첩 연산 → for 루프가 더 가독성이 좋을 수도 있음

---

🔗 참고 자료

• Bio-info 블로그
• ChatGPT