1. 引言

文件和目錄相關操作是軟件工程師必備的基本技能。
我不是說將一個文件復制到Windows文件資源管理器上的另一個文件夾中。
我是說使用代碼來進行自動化批量處理操作。
今天，我們將深入探討一個具體問題：

如何根據(jù)需要列出特定目錄下的所有文件名？

我將介紹5種用Python實現(xiàn)的方法。不同的方法適用于不同的使用場景。

2. 使用os.listdir()

使用Python中的os模塊下的方法listdir()是一個不錯的選擇，樣例如下：

>>> import os
>>> os.listdir()

上面的代碼將打印當前路徑下所有文件和目錄的名稱。
如果我們想基于特定路徑來打印結果，只需傳遞給函數(shù)os.listdir() 相應的參數(shù)，舉例如下：

>>> os.listdir(myPath)

如果我們只想打印所有文件不包含目錄，那么我們可以使用os.path.isfile() 來進行相應的過濾，如下：

>>> import os
>>> files = [f for f in os.listdir() if os.path.isfile(f)]

當然，對于目錄，同樣可以使用函數(shù)os.path.isdir() 進行過濾，代碼如下：

import os
files = [f for f in os.listdir() if os.path.isdir(f)]

如果我們只需要特定類型的文件怎么辦？
完全不用擔心。Python為復雜場景提供了更多選擇。

3. 使用os.walk()

os模塊中還有另一個方法 os.walk() 。顧名思義，它可以一層一層地“遍歷”目錄樹。當我們調(diào)用os.walk() 函數(shù)時，它將返回一個生成器。此時每次調(diào)用next() 方法生成下一個值時，它都會進入到一個layer ，結果是一個包含3個項的元組：(dirpath、dirname、filename) 。

舉例，如果要獲取第二層中所有文件夾的名稱，代碼如下所示：

from os import walk

f = []
layer = 1
w = walk("/Users/zhao")
for (dirpath, dirnames, filenames) in w:
? ? if layer == 2:
? ? ? ? f.extend(dirnames)
? ? ? ? break
? ? layer += 1

4. 在Glob模塊中使用正則表達式

有時我們不想獲取所有文件的名稱，而是希望精確獲取特定類型文件的名稱。
由于glob模塊能夠添加用于搜索的正則表達式，因此它將是我們進行此類操作的首選。

>>> import glob
>>> glob.glob("/sys/*.log")

上面的代碼將列出以log 結尾的文件名。

5. 使用pathlib

從Python 3.4開始，有一個名為pathlib 的模塊也很有用。
借助列表生成式的技巧，我們只需使用一行代碼即可生成當前路徑的所有文件名：

import pathlib

files = [f for f in pathlib.Path().iterdir() if f.is_file()]

令人奇怪的是，Path() 還附帶了glob() 函數(shù)。（無需在Python文件頂部顯式導入glob模塊）

import pathlib

files = [f for f in pathlib.Path().glob("/sys/*.log")]

6. 使用os.scandir()

經(jīng)典的os.listdir() 函數(shù)很直觀，但對于包含大量文件的大型目錄來說效率并不高。因此，Python 3.5引入了一個新的功能類似的函數(shù)os.scandir() 。
是的，有的同學可能已經(jīng)猜到了。此函數(shù)將返回一個生成器，并不是所有名稱的列表。在某些情況下，我們并不需要同時獲取所有名稱，此時會更有效。

>>> a=os.scandir()
>>> next(a)
<DirEntry 'test1.py'>
>>> next(a)
<DirEntry 'test2.py'>

7. 總結

本文重點介紹了在Python中如何列出目錄中的文件的五種方法，并針對每種方法給出了相應的代碼樣例。

到此這篇關于Python中列出目錄中的文件的五種方法小結的文章就介紹到這了,更多相關Python 列出目錄文件內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

最新評論