Membuat list directory dengan extension di python

Penulis bekerja dengan banyak data dalam proses pengujian suatu algoritma, misalkan saja dengan mudahnya meletakan ratusan file gambar kedalam satu folder kemudian diolah one by one untuk menghasilkan ciri fitur dan menyimpan nya dalam bentuk format csv. Tapi kali ini penulis hanya membahas cara me list suatu direktory/folder dengan sebuah extension, misakan saja untuk melihat suatu daftar file *.jpg
print daftar('.','jpg')
['A01.jpg', 'A02.jpg', 'A03.jpg', 'A04.jpg', 'A05.jpg', 'B01.jpg', 'B02.jpg', 'B03.jpg', 'B04.jpg', 'B05.jpg', 'C01.jpg', 'C02.jpg', 'C03.jpg', 'C04.jpg', 'C05.jpg']
Penulis sajikan code nya yaitu

import os
def daftar(folder,extension):
    nama_file = os.listdir(folder)
    m = list()
    index = 0
    for f in nama_file:
        po = str(f)
        if po[len(po)-len(extension):len(po)]==extension:
            m.insert(index,po)
            index = index +1
    return m