Python获取指定文件夹下的文件名

os.walk()和os.listdir()两种方法

一、os.walk()


模块os中的walk()函数可以遍历文件夹下所有的文件。

os.walk(top, topdown=Ture, onerror=None, followlinks=False)

该函数可以得到一个三元tupple(dirpath, dirnames, filenames).

  • dirpath:string,代表目录的路径;
  • dirnames:list,包含了当前dirpath路径下所有的子目录名字(不包含目录路径);
  • filenames:list,包含了当前dirpath路径下所有的非目录子文件的名字(不包含目录路径)。
    # -*- coding: utf-8 -*- 
    
    import os
    
    def file_name(file_dir): 
      for root, dirs, files in os.walk(file_dir):
      print(root) #当前目录路径
      print(dirs) #当前路径下所有子目录
      print(files) #当前路径下所有非目录子文件

当需要特定类型的文件时

# -*- coding: utf-8 -*- 

import os

def file_name(file_dir): 
  L=[] 
  for root, dirs, files in os.walk(file_dir):
    for file in files:
      if os.path.splitext(file)[1] == '.jpeg':
        L.append(os.path.join(root, file))
  return L

二、os.listdir()

os.listdir()函数得到的是仅当前路径下的文件名,不包括子目录中的文件,所有需要使用递归的方法得到全部文件名。

os.listdir(path)
----path -- 需要列出的目录路径

原文地址:https://www.cnblogs.com/soloate/p/10455329.html