本文采用os.walk()和os.listdir()两种方法,获取指定文件夹下的文件名。

python os模块获取指定文件夹下所有文件名

第一种方法使用os.walk():

os.walk() 方法用于通过在目录树种游走输出在目录中的文件名,向上或者向下。

语法

walk()方法语法格式如下:

os.walk(top[, topdown=True[, onerror=None[, followlinks=False]]])

参数

  •  top -- 根目录下的每一个文件夹(包含它自己), 产生3-元组 (dirpath, dirnames, filenames)【文件夹路径, 文件夹名字, 文件名】。

  •  topdown --可选,为True或者没有指定, 一个目录的的3-元组将比它的任何子文件夹的3-元组先产生 (目录自上而下)。如果topdown为 False, 一个目录的3-元组将比它的任何子文件夹的3-元组后产生 (目录自下而上)。

  •  onerror -- 可选,是一个函数; 它调用时有一个参数, 一个OSError实例。报告这错误后,继续walk,或者抛出exception终止walk。

  •  followlinks -- 设置为 true,则通过软链接访问目录。

样例代码:

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):  
  li=[]  
  for root, dirs, files in os.walk(file_dir): 
    for file in files: 
      if os.path.splitext(file)[1] == '.jpeg': 
        li.append(os.path.join(root, file)) 
  return li

其中os.path.splitext()函数将路径拆分为文件名+扩展名,例如os.path.splitext(“D:/.jpg”)将得到”D:/aaa“+".jpg"。

第二种方法使用os.listdir() 方法

os.listdir() 方法用于返回指定的文件夹包含的文件或文件夹的名字的列表。这个列表以字母顺序。 它不包括 '.' 和'..' 即使它在文件夹中。

语法

listdir()方法语法格式如下:

os.listdir(path)

参数

  • path -- 需要列出的目录路径

样例代码:

import os, sys

# 打开文件
path = "/var/"
dirs = os.listdir( path )

# 输出所有文件和文件夹
for file in dirs:
   print(file)

第三种使用列表推导式-获取每个文件的具体路径

import os

root="/home/chenxinming"

file_names=[os.path.join(path, name) for path, subdirs, files in os.walk(root) for name in files]

第四种只获取文件名

from os import listdir
from os.path import isfile, join

root="/home/chenxinming"
filenames = [f for f in listdir(root) if isfile(join(root, f))]

print(filenames [:10])


>>>['.profile', '.gitignore', '.mysql_history', '.bash_history',]

第五种更精简版本只获取文件名(只支持python3.x)

from os import walk

root="/home/chenxinming"

_, _, filenames = next(walk(root))

print(filenames [:10])

>>>['.profile', '.gitignore', '.mysql_history', '.bash_history',]


本文地址: http://www.chenxm.cc/article/220.html
版权声明: 本文为原创文章,版权归  陈新明  所有,欢迎分享本文,转载请保留出处!
上一篇: HTML介绍
下一篇: python leetcode 1. 两数之和
发表评论

还没有留言,还不快点抢沙发?