[python] 문자열이 목록에서 문자열 중 하나로 끝나는 지 확인하십시오.

PYTHON / 글쓴이

다음 코드를 작성하는 pythonic 방법은 무엇입니까?

extensions = ['.mp3','.avi']
file_name = 'test.mp3'

for extension in extensions:
    if file_name.endswith(extension):
        #do stuff

for루프 의 명시 적 선언을 피하고 if조건 에서 작성할 수 있다는 모호한 기억이 있습니다. 이것이 사실입니까?

답변

str.endswith 는 널리 알려져 있지는 않지만 튜플도 허용합니다. 반복 할 필요가 없습니다.

>>> 'test.mp3'.endswith(('.mp3', '.avi'))
True

답변

그냥 사용하십시오 :

if file_name.endswith(tuple(extensions)):

답변

파일에서 확장자를 가져 와서 확장자 세트에 있는지 확인하십시오.

>>> import os
>>> extensions = set(['.mp3','.avi'])
>>> file_name = 'test.mp3'
>>> extension = os.path.splitext(file_name)[1]
>>> extension in extensions
True

세트에서 조회의 시간 복잡성이 O (1) ( docs ) 이므로 세트를 사용합니다 .

답변

정규식과 문자열 (str) 메소드의 두 가지 방법이 있습니다.

문자열 메소드는 일반적으로 더 빠릅니다 (~ 2x).

import re, timeit
p = re.compile('.*(.mp3|.avi)$', re.IGNORECASE)
file_name = 'test.mp3'
print(bool(t.match(file_name))
%timeit bool(t.match(file_name)

루프 당 792ns ± 1.83ns (평균 ± 표준 7 회 실행, 각각 1000000 루프)

file_name = 'test.mp3'
extensions = ('.mp3','.avi')
print(file_name.lower().endswith(extensions))
%timeit file_name.lower().endswith(extensions)

루프 당 274ns ± 4.22ns (평균 7 번의 평균 ± 표준 편차, 각각 1000000 루프)

답변

나는 이것을 가지고있다:

def has_extension(filename, extension):

    ext = "." + extension
    if filename.endswith(ext):
        return True
    else:
        return False

답변

나는 다른 것을 찾는 동안 방금 이것을 만났습니다.

os패키지 의 메소드를 사용하는 것이 좋습니다 . 이상한 경우를 보완하여 더 일반적으로 만들 수 있기 때문입니다.

당신은 다음과 같은 것을 할 수 있습니다 : 

import os

the_file = 'aaaa/bbbb/ccc.ddd'

extensions_list = ['ddd', 'eee', 'fff']

if os.path.splitext(the_file)[-1] in extensions_list:
    # Do your thing.

답변

또 다른 가능성은 IN 문을 사용하는 것입니다.

extensions = ['.mp3','.avi']
file_name  = 'test.mp3'
if "." in file_name and file_name[file_name.rindex("."):] in extensions:
    print(True)