Python: os.listdir альтернатива / некоторые расширения

Question

Можно ли просматривать файлы с определенными расширениями с помощью команды os.listdir?Я хочу, чтобы он работал, поэтому он может показывать только файлы или папки с .f в конце.Я проверил документацию и ничего не нашел, так что не спрашивайте.

Answer 1

glob хорош в этом:

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

Answer 2

Не спрашивайте что?

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

Если вы хотите проверить список расширений, вы можете сделать очевидное обобщение,

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

или этот другой способ написать немного короче:

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

Answer 3

Существует еще одна возможность, не упомянутая до сих пор:

import fnmatch
import os

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

На самом деле, именно так реализован модуль glob, поэтому в этом случае glob проще и лучше, но модуль fnmatch может пригодиться в других ситуациях, например, при обходе дерева с помощью os.walk.

Answer 4

Попробуйте это:

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 ]

Answer 5

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