리스트 컴프리헨션의 다양한 사용
여태 [i for i in range()]
뿐이었던 나에게
range()
말고 다른것도 사용 가능하단걸 보여준 예.
1. 프로그래머스 python 코딩테스트
Level 1. 자리수 더하기
자연수 N이 주어지면, N의 각 자릿수의 합을 구해서 return 하는 solution 함수를 만들어 주세요.
예를들어 N = 123이면 1 + 2 + 3 = 6을 return 하면 됩니다.
정답 확인
[int(i) for i in str(n)]
def solution(n):
return sum([int(i) for i in str(n)])
sum
은 sum()
이거니까, list만드는건 []
.i
대신 리스트에 들어갈 건 문자 아닌 숫자 int(i)
여야 sum
이 가능함.range()
만 올 수 있는 줄 알았으나, str(n)
도 가능함.for c in 'hello'
뭐 이런것도 됨. c
는 i
대신 들어간 변형자. 쓰기 나름이니, for
를 잘 활용하자.