[Python] Usare glob con più estensioni

[fermat]

ciao!

per listare tutti i file con una certa estensione dentro ad una directory, ho usato glob.glob:

for infile in glob.glob(dir + "*.png"):
    # DO STUFF

però vorrei allargare la ricerca dei file anche ad altre estensioni.
ad esempio vorrei cercare tutti i png, jpeg, jpg.
come posso fare??

[clasku]

una cosa tipo questa?

extensions = ["*.png", "*.jpeg", "*.jpg"]
all_files = [glob.glob(ext) for ext in extensions]

[fermat]

ciao clasku!

allora, nella directory ci stanno tre immagini: due jpg e una png.
sto provando il codice che mi hai dato:

dir = "/home/matte/1_TEST/"
extensions = ["*.png", "*.jpeg", "*.jpg"]
all_files = [glob.glob(dir + ext) for ext in extensions]
print(all_files)

l'output è questo:

[['/home/matte/1_TEST/IMG2.png'], [], ['/home/matte/1_TEST/IMG.jpg', '/home/matte/1_TEST/TEST_JPG.jpg']]

quindi, le immagini nel trova tutte.
più c'è una voce vuota, e non capisco il perchè.

sennò ho trovato anche questo che pare funzionare:

extensions = [".png", ".jpeg", ".jpg"]
for infile in os.listdir(dir):
    ext = os.path.splitext(infile)[1]
    if ext.lower() in extensions:
        print(dir + infile)

ovviamente non usa glob, e controlla l'estensione del file ad ogni iterazione.

che ne dici??

[clasku]

la lista vuota è il glob per *.jpeg: glob restituisce sempre una lista, vuota se non trova corrispondenza

se vuoi, una volta che hai la lista dei file in forma di "lista di liste", puoi fare un'altra list comprehension per trasformarla in una lista semplice

lista = [elem for sublista in all_files for elem in sublista]

il secondo codice che hai messo non mi piace molto, oltre che credo sia più lento (ma questo valutalo tu, se vuoi)

[fermat]

in effetti sembra più veloce.
alla fine ho seguito il tuo codice (anche perchè di python non sono proprio esperto):

extensions = ["*.png", "*.jpeg", "*.jpg"]
allFiles = [glob.glob(dir + ext) for ext in extensions]
lista = [elem for sublista in allFiles for elem in sublista]
for infile in lista:
    try:
        file, ext = os.path.splitext(infile)
        im = Image.open(infile)
        im.thumbnail(size, Image.ANTIALIAS)
        im.save(file + "_R" + ext)
        print(file)
    except IOError as e:
        print(e)

questo ha funzionato alla perfezione!
grazie per la dritta!