[python] 리스트 함축(comprehension)

• 리스트 함축은 코드를 짧고 간결하게 만드는데 사용된다.
• 리스트 함축을 사용하면 간결하면서도 효율적으로 리스트를 생성할 수 있다.
• 입력으로는 리스트, 튜플, 문자열등의 시퀀스만 가능하다.

[x*x for x in range(10)]  # 0에서 9까지 숫자의 제곱 값
# [0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

st = 'Hello World'
[x.upper() for x in st]  # 문자열 각각에 대해 upper() 메서드 적용
# ['H', 'E', 'L', 'L', 'O', ' ', 'W', 'O', 'R', 'L', 'D']

a = ['welcome', 'to', 'the', 'python', 'world']
first_a = [s[0].upper() for s in a]
first_a
# ['W', 'T', 'T', 'P', 'W']

• 조건이 붙는 리스트 함축표현도 가능하다.

s = ['Hello', '12345', 'World', '67890']
numbers = [x for x in s if x.isdigit()]  # 숫자로만 이루어진 문자열을 필터링

print(numbers)
# ['12345', '67890']