获取文件夹下所有文件名

    In [1]:
    import os
    from pathlib import Path
    import glob
    
    
      
      
      
      
    

    In [3]:
    dirpath="D:\ test\ 1"
    #判断是否是目录
    if os.path.isdir(dirpath):
    img_names=[]
    # 获取所有文件名
    dir_names = os.listdir(dirpath)
    for img_name in dir_names:
        # 获取固定后缀文件名
        if os.path.splitext(img_name)[1] == '.jpg':
            img_names.append(img_name)
    else:
    print("路径错误，请输入目录！")
    
    print("dir_names:",dir_names)
    print("img_names:",img_names)
    print("os.path.splitext(img_name):",os.path.splitext(img_names[0]))
    
    
      
      
      
      
      
      
      
      
      
      
      
      
      
      
      
      
      
    

    In [4]:
    # 获取所有文件路径
    file_paths=glob.glob(dirpath)
    # 获取固定后缀文件路径
    img_paths=glob.glob(dirpath+"\ *.jpg")
    print("file_paths:",file_paths)
    print("img_paths:",img_paths)
    
    
      
      
      
      
      
      
      
    

    In [5]:
    # 获取固定后缀文件路径
    image_suffixes = (".jpeg", ".jpg", ".png", ".bmp")
    img_paths=[p for p in Path(dirpath).glob("**/*") if p.suffix.lower() in image_suffixes]
    print("img_paths:",img_paths)
    print("img_paths[0]:",img_paths[0])