获取文件夹名 python

import os

path = os.getcwd()
print(path)
test_path = os.path.join(path,"validation")

for root,dirs,files in os.walk(test_path):
    for file in files:
        if '.mp4' in file:
            sub_path = os.path.join(test_path, file)
            dir1 = root.replace(test_path,"")
            label = dir1.replace("\\","")
            print('文件夹\n',label)
            print(sub_path)
            print("文件 \n",file)

其中主要是我想要把文件夹名作为实际的标签值，所以必须知道文件所在文件夹的名字，我主要是实现对视频文件对应的文件夹读取，如果想要是图片，可以直接改成对应的jpg后缀

import os
from utils import extract_frames, load_frames, render_frames


path = os.getcwd()
print(path)
test_path = os.path.join(path,"validation")
lis = []
for root,dirs,files in os.walk(test_path):
    if dirs:
        labels = dirs
        print(dirs)
    for file in files:
        if '.mp4' in file:
            sub_path = os.path.join(root, file)
            # file_name = os.path.basename(sub_path)
            label = sub_path.replace(test_path,"")
            label = label[1:]#去除\\，其主要代表一个字符
            #frames = extract_frames(file)
            lis.append(label)
            print('文件夹\n',label)
            print(sub_path)
            print("文件 \n",file)

第二种方法，就是把字符串删除掉“\”，其代表一个字符label = label[1:]
参考：python 去除字符串（去除前面几个或者是后面几个）