본문 바로가기

COMPUTER SCIENCE/PYTHON

[Python] itemgetter를 이용하여 index로 list 추출하기

 

from operator import itemgetter

a = [1,2,3,4,5,6]
list_index = [1,3,5]

itemgetter(*list_index)(a)
>>> [2,4,6]

 

반응형