Импорт определенных файлов с использованием glob в Python

У меня есть несколько файлов, и я хочу импортировать их в определенные массивы на основе части их имени.

dark_1s-001.fit
dark_1s-002.fit
dark_1s-003.fit

dark_5s-001.fit
dark_5s-002.fit
dark_5s-003.fit

Мой единственный реальный вопрос заключается в том, как мне изменить аргументы из glob.glob('*.fit'), чтобы различать атрибуты файла «1s» и «5s».

files_1s = glob.glob('*.fit')
files_5s = glob.glob('*.fit')

darks_1s = []
darks_5s = []

for f1, f2 in zip(files_1s, files_5s):
    darks_1s.append(fits.getdata(f1))
    darks_5s.append(fits.getdata(f1))

darks_1s = np.array(darks_1s)
darks_5s = np.array(darks_5s)

median_dark_1s = np.median(darks_1s, axis=0)
median_dark_5s = np.median(darks_5s, axis=0)

person Bobby Stiller    schedule 10.12.2017    source источник


Ответы (1)


Вы можете вставлять определенные части имен файлов, которые должны существовать, в строку glob.

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

files_1s = glob.glob('*1s-*.fit')
files_5s = glob.glob('*5s-*.fit')
person James    schedule 10.12.2017