如何使用子目录的数据文件并在Python中执行迭代操作

Question

我在当前目录中有我的jupyter笔记本（Python脚本）。在当前目录中，我有两个子文件夹，即a和b。在两个目录中a和b我的.dat文件具有相同名称。例如，目录a包含文件，即x1-x1-val_1，x1-x1-val_5，x1-x1-val_11 ... x1-x1-val_86和x1-x2-val_1，x1-x2-val_5，x1 -x2-val_11 ... x1-x2-val_86，IE值在范围内（1,90,5）。同样，我有目录中的文件b。

我想使用我的python脚本访问a和b中的文件，以在.dat文件上执行迭代操作。我现在的代码仅在当前目录中保存目录 a a a a b 的文件时才有效。例如，我的脚本使用以下函数。

def get_info(test):     
    my_dict = {'test':test}   
    c = []    
    
    for i in range(1,90,5):    
        x_val = 'x_val_'+test+'-val_'+str(i)
        y_val = 'y_val_'+test+'-val_'+str(i)
        
        my_dict[x_val],my_dict[y_val]= np.loadtxt(test+'-val_'+str(i)+'.dat'
                                                        ,usecols= (1,2),unpack=True)
        dw = compute_yy(my_dict[x_val],my_dict[y_val],test)
        c.append(dw)
    
    my_dict.update({test+'_c'+:np.array(c)})

    return my_dict

我调用get_info（）通过使用以下方式：

tests = ['x1-x1', 'x1-x2']

new_dict = {}
for i in tests:
     new_dict.update({i:get_info(i)})

如何使用代码在Directory a和/或b中访问文件？我知道这是关于提供正确的道路，但我不确定该怎么做。我想的一种方法是遵循；

ext = '.dat'
for files in os.listdir(path_to_dir):
    if files.endswith(ext):
        print(files)   # do operations

替代方法可能是使用os.path.join（）。但是，我无法解决它，因此我可以使用相同的python脚本（也许有最小的更改）可以使用文件并在子文件夹中迭代a a and b 。感谢您提前的反馈！

Answer 1

如果要在每个文件夹sameatelly上运行get_info（），则有两种方法：

首先：由 @Medium-difimensional在评论中描述，

您可以使用os。 CHDIR（文件夹）更改当前工作目录。然后，代码将在此文件夹中使用文件运行，

中查看当前工作目录

os.chdir("a")
get_info(i)
os.chdir("..") # move back to parent folder

os.chdir("b")
get_info(i)
os.chdir("..") # move back to parent folder

您可以在print（os.getCwd（）） chdir（）（类似于命令cd）在控制台中）可以使用相对路径（r“ a“）完整路径（r“ c：\ full \ path \ to \ a”）和<代码> .. 移动到父文件夹（r“ a \ .. \ b”）

如果文件可以在嵌套文件夹中，则可能不会回去您可以使用getcwd（）

cwd = os.getcwd()

os.chdir("folder1/folder2/a")
get_info(i)
os.chdir(cwd) # move back to previous folder

os.chdir("folder1/folder2/b")
get_info(i)
os.chdir(cwd) # move back to previous folder

（btw：在Linux上的控制台您可以使用cd-移动到以前的文件夹）

second：当您打开文件时，请使用文件夹，

每个获取文件名的命令也可以使用文件夹\ filename获得路径（可以是相对路径，完整路径和..）喜欢

• r“ a \ filename.dat”
• r“ c：\ full \ path \ to \ b \ filename.dat”
• r“ a \ .. \ b \ filename.dat“，

因此您可以使用Extra option 文件夹定义函数

def get_info(text, folder):

，并在读取文件

loadtxt(folder + r'\' + test+'-val_'+str(i)+'.dat', ...)

或使用f-string的更读取

loadtxt(rf'{folder}\{test}-val_{i}.dat', ...)

时使用此文件夹，然后再使用此文件 夹。将其运行为

get_info(i, "a")

get_info(i, "b")