itemgetter 사용하기

sort문제를 풀 때 key=lambda x:x[1] 이런식으로 풀었었는데 내 기준으로 이것보다 간단한 itemgetter 를 알게되었다.

from operator import itemgetter를 써주면 사용할 수 있다.

from operator import itemgetter
data = [ ("gg", 25, 11), ("sd", 342, 972), ("aa", 36, 616), ("td", 16, 234), ("oi", 95, 435) ]

data.sort(key=itemgetter(0))
print(data) #[('aa', 36, 616), ('gg', 25, 11), ('oi', 95, 435), ('sd', 342, 972), ('td', 16, 234)]

 

from operator import itemgetter
data = [ ("gg", 25, 11), ("sd", 342, 972), ("aa", 36, 616), ("td", 16, 234), ("oi", 95, 435) ]

data.sort(key=itemgetter(1))
print(data) #[('td', 16, 234), ('gg', 25, 11), ('aa', 36, 616), ('oi', 95, 435), ('sd', 342, 972)]

 

from operator import itemgetter
data = [ ("gg", 25, 11), ("sd", 342, 972), ("aa", 36, 616), ("td", 16, 234), ("oi", 95, 435) ]

data.sort(key=itemgetter(2))
print(data) #[('gg', 25, 11), ('td', 16, 234), ('oi', 95, 435), ('aa', 36, 616), ('sd', 342, 972)]

너무 편하다,,,,,,,,,,, 왜 이제 알았을까 라는 생각뿐,,,,,,,,

2020/05/31 - [개발이야기/python] - itemgetter 사용하기

itemgetter 사용하기