在 Python 中使用 glob 导入特定文件

Question

我有一些文件，我想根据部分名称将它们导入特定的数组。

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)

Answer 1

您可以将必须​​存在的文件名的特定部分夹在 glob 字符串中。

试试这个：

files_1s = glob.glob('*1s-*.fit')
files_5s = glob.glob('*5s-*.fit')