使用 Python 在循环中使用生成器读取多个 txt 文件

Question

程序调用该函数并在循环中只打开一次。你有什么解决方案可以让它一直调用循环中的函数吗？ （我的意思是每个 j）

def list_spectra(interpolated, txt):
    return (f for f in listdir(interpolated) if f.endswith("." + txt))

wl = []
fl = []
er = []

flux  = []
error = []

med_flux = []
med_err  = []

spectra_all = list_spectra("C:\\..my path here..\\interpolated","txt")
path = "C:\\..my path here..\\interpolated\\"

for j in range(0,len(wavelength)):
   for f in spectra_all:
        data = np.loadtxt(path+f)
        for ii in range(0,len(data)):

            wl = data[:,0]
            fl = data[:,1]
            er = data[:,2]    

            for i in range(0,len(wl)):
                if wl[i] == wavelength[j]:
                    flux.append(fl[i])
                    error.append(er[i])

    med_flux.append(np.median(flux))
    med_err.append(np.median(error))

Answer 1

你是这个意思吗：

for j in range(0,len(wavelength)):
    spectra_all = list_spectra("C:\\..my path here..\\interpolated","txt")
    for f in spectra_all:
        data = np.loadtxt(path+f)
        # etc