如何获取每个文件的目录名？[副本]

Question

root
--- audi    (dirs)
           ---11 01012020.csv  (files)
           ---01 102020.csv
--- bmw
            ---66  10052020.csv
            ---43  11112020.csv
--- mercedes
             ---34  21062020.csv
             ---23  30112020.csv

上面是我的根、目录和路径的结构。我想得到三样东西<代码>汽车，文件id和日期。

1. Car是目录的名称

这是我的代码：

for root, dirs, files in os.walk(path_root):
    for file in files:
        if file.endswith('.csv'):
            file_id = file.split()[0]
            date = re.search(' (\d+).', file).group(1)
            car = ? idk how to get this one.

如您所见，我设法获得了文件id和日期。现在我想得到每个文件的车名（directroy的名字）。。。最简单的方法是什么？

Answer 1

root变量包含当前目录；你想要这辆车的最后一个部件。

for root, dirs, files in os.walk(path_root):
    for file in files:
        if file.endswith('.csv'):
            file_id = file.split()[0]
            date = re.search(' (\d+).', file).group(1)
            car = os.path.basename(root)

如果您的文件名具有代表性，那么您实际上不需要正则表达式来提取日期。可能file.split（）[1]。split（'..'）[0]