본문 바로가기

Python

[python] 리스트 문자열 중 길이가 가장 짧은 문자열 찾기

반응형

 

단순 루프 돌면서 찾기

strs = ["banana", "apple", "watermelon", "strawberry"]

min_str = strs[0]
for i in range(1, len(strs)):
     if len(min_str) > len(strs[i]):
       min_str = strs[i]
print(min_str)

>> apple

 

보통 리스트에서 가장 짧은 문자열을 찾을때는 단순 루프를 돌면서 데이터를 찾았다. 

그런데 leetcode 문제 풀이중 훨씬 간단한 방법으로 이를 찾는 방법을 찾았다. 

 

 

min 함수 이용하기

strs = ["banana", "apple", "watermelon", "strawberry"]

min_str = min(strs, key=len)

print(min_str)

min 함수 내부에 key로 len 함수를 전달해주면 Iterable한 자료형의 원소에 대해 해당 함수를 적용한 값을 토대로 min 값을 찾아준다. 만약 max 값을 찾고 싶다면 max 함수를 쓰면된다.

 

 

반응형