ตรวจสอบว่าสตริงลงท้ายด้วยหนึ่งในสตริงจากรายการ

วิธีการเขียนแบบต่อเนื่องของไพ ธ อนคืออะไร?

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

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

ฉันมีหน่วยความจำที่คลุมเครือว่าการประกาศforลูปที่ชัดเจนสามารถหลีกเลี่ยงและเขียนในifสภาพ มันเป็นเรื่องจริงเหรอ?

แม้ว่าจะไม่เป็นที่รู้จักอย่างแพร่หลายstr.endswithก็ยอมรับ tuple เช่นกัน คุณไม่จำเป็นต้องวนซ้ำ

>>> '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) ( เอกสาร )

มีสองวิธีคือนิพจน์ทั่วไปและวิธีสตริง (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)

792 ns ± 1.83 ns ต่อวง (หมายถึง± std. dev. ของ 7 วิ่ง 1000000 ลูปแต่ละอัน)

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

274 ns ± 4.22 ns ต่อวง (หมายถึง± std. dev. จาก 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)

อีกวิธีหนึ่งซึ่งสามารถส่งคืนรายการของสตริงที่ตรงกันได้

sample = "alexis has the control"
matched_strings = filter(sample.endswith, ["trol", "ol", "troll"])
print matched_strings
['trol', 'ol']