xs = [x.data for x in inputs]
이런 형태의 코드를 본 적이 있을 것이다.
위 코드는 inputs 리스트의 각 원소 x에 대해 각각의 데이터 x.data를 꺼내고, 꺼낸 원소들로 구성된 새로운 리스트를 만든다.
이러한 형태를 list comprehension, 우리말로 리스트 내포라고 말한다.
list compreshension은 python코드가 간결하게 표현되는 강력한 무기 중 하나로 코딩테스트의 모범 답안에서 자주 볼 수 있다.
다만 새로운 리스트가 다시 생성되므로 메모리 효용성에는 주의해야 할 것이다.
이 때 순서는 좀 중요한다.
반드시 아래 형태를 띄어야 한다.
[ 조건 만족 시 출력값 if 조건 else 조건 불만족 시 출력 값 for i in data]
반응형
'전공 > python' 카테고리의 다른 글
PyTorch module forward() 와 특수 매소드 __call__() 비교 (3) | 2022.07.12 |
---|---|
python size function (0) | 2022.06.23 |
heap을 쓸 수 있는 상황이면 무조건 써라 (0) | 2022.06.14 |
[Error 해결] KeyError: '4' (0) | 2022.06.10 |
데이터 전처리 중 오류들 (0) | 2022.04.23 |
댓글