Python - 使用通配符获取使用 glob 打开的文件的当前目录

Question

我正在使用 glob() 来读取目录中的所有文件。我需要读取所有子文件夹中的每个 csv 文件。在我阅读并处理每个文件后，我想将每个新文件保存在另一个文件夹中，但保留子文件夹名称。打开我使用的文件：

data = "data/**/*.csv"
csv_files = glob(data, recursive=True)

然后我读取每个文件：

for f in csv_files:
        file = pd.read_csv(f, sep=';')
        
        fileName = os.path.basename(f)
        path = os.path.dirname(data)
        fileFolder = os.path.basename(path)

#       Do some stuff
#
#       Then save the file

        file.to_csv("new_data/"+fileFolder+fileName)

我的问题是 fileFolder 返回 ** 而不是实际的文件夹名称

我的源目录如下：

data / 
       - January 2021
       - February 2021
             .
             .
             .
       - December 2021

我想把我的文件保存在

new_data / 
       - January 2021
       - February 2021
             .
             .
             .
       - December 2021

我正在寻找一种不需要手动创建包含所有子文件夹的列表的方法。

Answer 1

试试这个：

for f in csv_files:
    file = pd.read_csv(f, sep=';')
        
#    Do some stuff
#
#    Then save the file

    new_path = os.path.join(*f.split(os.path.sep)[1:])
    file.to_csv(os.path.join('new_data', *new_path))

它是做什么的

1. 在 macOS/Linux 上按 / 和在 Windows 上按 \ 拆分文件路径
2. 删除第一个（最顶层目录）
3. 与'new_data' 一起重新加入它