Periksa apakah string berakhir dengan salah satu string dari daftar

220

Apa cara pythonic dalam menulis kode berikut?

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

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

Saya memiliki memori yang tidak jelas bahwa deklarasi eksplisit dari forloop dapat dihindari dan ditulis dalam ifkondisi tersebut. Apakah ini benar?

python string list

— TheMeaningfulEngineer
sumber

2

Meskipun pertanyaan ini dijawab dengan baik, mungkin penulis awalnya memikirkannya if any((file_name.endswith(ext) for ext in extensions)).

— sapht

450

Meskipun tidak dikenal secara luas, str.endswith juga menerima tuple. Anda tidak perlu mengulang.

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

— falsetru
sumber

10

apakah Anda tahu mengapa itu tidak akan menerima daftar tetapi apakah tuple? hanya ingin tahu

— ilyail3

2

@ falsetru Tautan dalam jawaban tidak secara eksplisit menjawab pertanyaan itu. Itu hanya menyebutkan bahwa ia dapat menerima tupel, tetapi tidak mengapa ia tidak dapat menerima daftar. Karena keduanya merupakan urutan, satu-satunya perbedaan yang berpotensi saya lihat adalah daftar dapat berubah, sementara tupel tidak dapat diubah. Saya mungkin salah, tetapi saya tidak bisa melihat alasan lain mengapa itu dinyatakan secara eksplisit.

— KymikoLoco

4

Jika Anda ingin memeriksa apakah sebuah string diakhiri dengan surat:import string; str.endswith(tuple(string.ascii_lowercase))

— Alex Willison

3

hanya sebuah catatan, endswithterima tuple hanya untuk python 2.5 ke atas

— Akash Singh

1

Tidak pernah tahu ini! Itu sempurna!

— fool4jesus

46

Cukup gunakan:

if file_name.endswith(tuple(extensions)):

— Jon Clements
sumber

Sederhana dan efektif!

— Ced

6

Ambil ekstensi dari file dan lihat apakah ekstensi itu ada di set ekstensi:

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

Menggunakan set karena kompleksitas waktu untuk pencarian dalam set adalah O (1) ( docs ).

— alecxe
sumber

8

Hanya untuk catatan seperti yang Anda sebutkan efisiensi, untuk tupel cukup singkat, yang .endswith()dengan tupel diinternir akan lebih cepat dari satu set lookup

— Jon Clements

@JonClements Saya pikir Anda perlu lencana komentar emas SO khusus untuk membuat catatan yang luar biasa pada jawaban dan pertanyaan :)

— alecxe

Nah - aku hanya akan untuk "Stalking alecxe" lencana;)

— Jon Clements

2

Perhatikan juga bahwa pada 2.7 dan yang lebih baru, Anda dapat menggunakan sintaks matematika untuk set,, {'.mp3','.avi'}ia menghindari konversi tipe tambahan dan mungkin lebih mudah dibaca tergantung pada latar belakang Anda ('Meskipun dapat menyebabkan kebingungan dengan kamus, dan tidak dapat digunakan untuk membuat kosong set).

— Perkins

@JonClements suatu hari nanti aku akan menjadi sama bijaknya denganmu :)

— alecxe

3

Ada dua cara: ekspresi reguler dan metode string (str).

Metode string biasanya lebih cepat (~ 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 per loop (rata-rata ± std. Dev dari 7 run, masing-masing 10.00000 loop)

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

274 ns ± 4,22 ns per loop (rata-rata ± std. Dev dari 7 run, masing-masing 10.00000 loop)

— Igor A
sumber

1

Aku punya ini:

def has_extension(filename, extension):

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

— Thomas Wouters
sumber

1

Maksudmu return filename.endswith(ext)? : P

— Mr_and_Mrs_D

1

Saya baru saja menemukan ini, sambil mencari sesuatu yang lain.

Saya akan merekomendasikan untuk pergi dengan metode dalam ospaket. Ini karena Anda bisa membuatnya lebih umum, mengimbangi kasus aneh apa pun.

Anda dapat melakukan sesuatu seperti:

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.

— Xxxo
sumber

0

Kemungkinan lain bisa menggunakan pernyataan IN:

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

— Tidak ada yang hilang
sumber

@ Rainald62, indexseharusnya rindexdalam kasus itu.

— NeverHopeless

0

Cara lain yang dapat mengembalikan daftar string yang cocok adalah

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

— Akash Singh
sumber