在python中使用更改读取新文件

Question

我正在使用import os库来处理当前目录中的所有文件夹。但是，每当程序开始遍历目录中的新文件夹时，我都需要程序请求文件名或路径。示例:假设目录中有两个文件夹，即：

1. 电平控制
2. 电平更改.

当程序通过电平控制文件夹时，它应该读取文件a.txt。在遍历“级别更改”文件夹时，应该读取文件b.txt。还假设这两个文件位于同一个目录中。

Answer 1

通过查看您的描述，下面的代码应该可以工作(Python3.x)，但是在描述中没有提到一些情况(例如，如何处理文件夹和文件)，因此您可能需要做一些修改。

import os

def walk(root):
    dirs = []
    for item in os.listdir(root):
        if os.path.isdir(os.path.join(root,item)):
            dirs.append(item)
    if len(dirs) > 0:
        for n,d in enumerate(dirs):
            print(f'[{n}] : {d}')
        folder = input('which folder ? ')
        path = os.path.join(root, folder)
        if os.path.isdir(path):
            walk(path)
        print('Invalid path')
    else:
        for file in os.listdir(root):
            print(f'read : {file}')
    
walk('.')

您可以使用os.walk()代替os.listdir()，因为您需要根据当前目录中的内容执行操作。