Как читать только файлы wav в каталоге с помощью Python?

Question

from scipy.io.wavfile import read 
files = [f for f in os.listdir('.') if os.path.isfile(f)] 
print files 
for i in range(0,1): 
w = read(files[i]) 
print w 

Мне нужно только прочитать .wav-файлы из рабочего каталога python. И сохраните каждый .wav-файл в виде массива numpy. Это мой код. Но в этом коде все файлы читаются. Я только читаю wav-файлы в каталоге? Как это возможно?

Answer 1

Используйте модуль glob и передайте ему шаблон (например, *.wav или любые названия файлов). Он вернет список файлов, соответствующих критериям или шаблону.

import glob 
from scipy.io.wavfile import read 

wavs = [] 
for filename in glob.glob('*.wav'): 
    print(filename) 
    wavs.append(read(filename)) 

Answer 2

Если вы не хотите использовать glob, то что-то, как это будет работать:

files = [f for f in os.listdir('.') if os.path.isfile(f) and f.endswith(".wav")]