Python獲取指定文件夾下的文件名的方法

更新時間：2018年02月06日 13:55:24 作者：小路孩

本篇文章主要介紹了Python獲取指定文件夾下的文件名的方法，本文采用os.walk()和os.listdir()兩種方法，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧

本文采用os.walk()和os.listdir()兩種方法，獲取指定文件夾下的文件名。

一、os.walk()

模塊os中的walk()函數(shù)可以遍歷文件夾下所有的文件。

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

該函數(shù)可以得到一個三元tupple(dirpath, dirnames, filenames).

參數(shù)含義：

dirpath：string，代表目錄的路徑；
dirnames：list，包含了當(dāng)前dirpath路徑下所有的子目錄名字（不包含目錄路徑）；
filenames：list，包含了當(dāng)前dirpath路徑下所有的非目錄子文件的名字（不包含目錄路徑）。

注意，dirnames和filenames均不包含路徑信息，如需完整路徑，可使用os.path.join(dirpath, dirnames)

下面給出代碼;

# -*- coding: utf-8 -*-  
import os 
def file_name(file_dir):  
  for root, dirs, files in os.walk(file_dir): 
    print(root) #當(dāng)前目錄路徑 
    print(dirs) #當(dāng)前路徑下所有子目錄 
    print(files) #當(dāng)前路徑下所有非目錄子文件

當(dāng)需要特定類型的文件時，代碼如下：

# -*- 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.path.splitext()函數(shù)將路徑拆分為文件名+擴(kuò)展名，例如os.path.splitext(“E:/lena.jpg”)將得到”E:/lena“+".jpg"。

二、os.listdir()

os.listdir()函數(shù)得到的是僅當(dāng)前路徑下的文件名，不包括子目錄中的文件，所有需要使用遞歸的方法得到全部文件名。

直接給出代碼，函數(shù)將返回類型為‘.jpeg'個文件名：

# -*- coding: utf-8 -*- 
import os 
 
def listdir(path, list_name): 
  for file in os.listdir(path): 
    file_path = os.path.join(path, file) 
    if os.path.isdir(file_path): 
      listdir(file_path, list_name) 
    elif os.path.splitext(file_path)[1]=='.jpeg': 
      list_name.append(file_path)

初學(xué)python，犯了些低級錯誤，浪費(fèi)了很多時間，首先一定要確保代碼的四空格縮進(jìn)，注意直接回車下一行并不是四空格縮進(jìn)；還有一點(diǎn)是在命令提示符窗口（Windows）使用python解釋器第一次運(yùn)行代碼后，修改源代碼，如若再次運(yùn)行需要退出重新進(jìn)入python解釋器，否則仍將運(yùn)行第一次執(zhí)行過的代碼。

以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: