Python: alternativa os.listdir/alcune estensioni

Question

È possibile visualizzare file con determinate estensioni con il comando os.listdir? Voglio che funzioni, in modo che possa mostrare solo file o cartelle con .f alla fine. Ho controllato la documentazione e non ho trovato nulla, quindi non chiedere.

Answer 1

glob è bravo in questo:

import glob 
for f in glob.glob("*.f"): 
    print f 

Answer 2

Non chiedere cosa?

[s for s in os.listdir() if s.endswith('.f')] 

Se si desidera controllare un elenco di estensioni, si potrebbe fare la generalizzazione ovvia,

[s for s in os.listdir() if s.endswith('.f') or s.endswith('.c') or s.endswith('.z')] 

o quest'altro modo è un po 'più corto di scrivere:

[s for s in os.listdir() if s.rpartition('.')[2] in ('f','c','z')] 

Answer 3

[s for s in os.listdir() if os.path.splitext(s) == 'f'] 

Answer 4

C'è un'altra possibilità non menzionata finora:

import fnmatch 
import os 

for file in os.listdir('.'): 
    if fnmatch.fnmatch(file, '*.f'): 
     print file 

In realtà questo è il modo è implementato il modulo glob, quindi in questo caso glob è più semplice e meglio, ma il modulo fnmatch può essere utile in altri situazioni, ad es quando si fa un attraversamento dell'albero usando os.walk.

Answer 5

Prova questo:

from os import listdir 

extension = '.wantedExtension' 

mypath = r'my\path' 

filesWithExtension = [ f for f in listdir(mypath) if f[(len(f) - len(extension)):len(f)].find(extension)>=0 ]