本期責(zé)編：Sophie

原文來源 | devshed

譯文來源 | wwwjfy

一、Python中的文件管理

文件管理是很多應(yīng)用程序的基本功能和重要組成部分。Python可以使文件管理極其簡(jiǎn)單，特別是和其它語言相對(duì)比。

以下，Peyton McCullough講解了文件管理基礎(chǔ)。

介紹

你玩過的游戲使用文件來保存存檔；你下的訂單保存在文件中；很明顯，你早上寫的報(bào)告也保存在文件中。

幾乎以任何語言編寫的眾多應(yīng)用程序中，文件管理是很重要的一部分。Python當(dāng)然也不例外。在這篇文章中，我們將探究如何使用一些模塊來操作文件。我們會(huì)完成讀文件，寫文件，增加文件內(nèi)容的操作，寫文件，增加文件內(nèi)容的操作，還有一些另類的用法。OK，讓我們開始吧。

讀寫文件

最基本的文件操作當(dāng)然就是在文件中讀寫數(shù)據(jù)。這也是很容易掌握的。現(xiàn)在打開一個(gè)文件以進(jìn)行寫操作：

1 fileHandle = open ( 'test.txt'， 'w' )

‘w’是指文件將被寫入數(shù)據(jù)，語句的其它部分很好理解。下一步就是將數(shù)據(jù)寫入文件：

1 fileHandle.write ( 'This is a test./nReally， it is.' )

這個(gè)語句將“This is a test.”寫入文件的第一行，“Really， it is.”寫入文件的第二行。最后，我們需要做清理工作，并且關(guān)閉文件：

1 fileHandle.close()

正如你所見，在Python的面向?qū)ο髾C(jī)制下，這確實(shí)非常簡(jiǎn)單。需要注意的是，當(dāng)你再次使用“w”方式在文件中寫數(shù)據(jù)，所有原來的內(nèi)容都會(huì)被刪除。如果想保留原來的內(nèi)容，可以使用“a”方式在文件中結(jié)尾附加數(shù)據(jù)：

1 fileHandle = open ( 'test.txt'， 'a' )

2 fileHandle.write ( '/n/nBottom line.' )

3 fileHandle.close()

然后，我們讀取test.txt，并將內(nèi)容顯示出來：

1 fileHandle = open ( 'test.txt' )

2 print fileHandle.read()

3 fileHandle.close()

以上語句將讀取整個(gè)文件并顯示其中的數(shù)據(jù)。我們也可以讀取文件中的一行：

1 fileHandle = open ( 'test.txt' )

2 print fileHandle.readline() # 'This is a test.'

3 fileHandle.close()

同時(shí)，也可以將文件內(nèi)容保存到一個(gè)list中：

1 fileHandle = open ( 'test.txt' )

2 fileList = fileHandle.readlines()

3 for fileLine in fileList：

4 print '>>'， fileLine

5 fileHandle.close()

Python在讀取一個(gè)文件時(shí)，會(huì)記住其在文件中的位置，如下所示：

1 fileHandle = open ( 'test.txt' )

2 garbage = fileHandle.readline()

3 fileHandle.readline() # 'Really， it is.'fileHandle.close()

可以看到，只有第二行顯示出來。然而，我們可以讓Python從頭開始讀來解決這個(gè)問題：

1 fileHandle = open ( 'test.txt' )

2 garbage = fileHandle.readline()

3 fileHandle.seek ( 0 )

4 print fileHandle.readline() # 'This is a test.'

5 fileHandle.close()

在上面這個(gè)例子中，我們讓Python從文件第一個(gè)字節(jié)開始讀取數(shù)據(jù)。所以，第一行文字顯示了出來。當(dāng)然，我們也可以獲取Python在文件中的位置：

1 fileHandle = open ( 'test.txt' )

2 print fileHandle.readline() # 'This is a test.'

3 print fileHandle.tell() # '17'

4 print fileHandle.readline() # 'Really， it is.'

或者在文件中一次讀取幾個(gè)字節(jié)的內(nèi)容：

1 fileHandle = open ( 'test.txt' )

2 print fileHandle.read ( 1 ) # 'T'

3 fileHandle.seek ( 4 )

4 print FileHandle.read ( 1 ) # ' '(原文有錯(cuò))

在Windows和Macintosh環(huán)境下，有時(shí)可能需要以二進(jìn)制方式讀寫文件，比如圖片和可執(zhí)行文件。此時(shí)，只要在打開文件的方式參數(shù)中增加一個(gè)“b”即可：

1 fileHandle = open ( 'testBinary.txt'， 'wb' )

2 fileHandle.write ( 'There is no spoon.' )

3 fileHandle.close()

1 fileHandle = open ( 'testBinary.txt'， 'rb' )

2 print fileHandle.read()

3 fileHandle.close()

二、從現(xiàn)有文件中獲取信息

使用Python中的模塊，可以從現(xiàn)有文件中獲取信息。使用“os”模塊和“stat”模塊可以獲取文件的基本信息：

1 import os

2 import stat

3 import time

4

5 fileStats = os.stat ( 'test.txt' )

6 fileInfo = {

7 'Size' ： fileStats [ stat.ST_SIZE ]，

8 'LastModified' ： time.ctime ( fileStats [ stat.ST_MTIME ] )，

9 'LastAccessed' ： time.ctime ( fileStats [ stat.ST_ATIME ] )，

10 'CreationTime' ： time.ctime ( fileStats [ stat.ST_CTIME ] )，

11 'Mode' ： fileStats [ stat.ST_MODE ]

12 }

13

14 for infoField， infoValue in fileInfo：

15 print infoField， '：' + infoValue

16 if stat.S_ISDIR ( fileStats [ stat.ST_MODE ] )：

17 print 'Directory. '

18 else：

19 print 'Non-directory.'

上面這個(gè)例子創(chuàng)建了一個(gè)包含文件基本信息的dictionary。然后顯示了相關(guān)信息，并且告訴我們打開的是否為目錄。我們也可以試一下打開的是否是其它幾種類型：

1 import os

2 import stat

3

4 fileStats = os.stat ( 'test.txt' )

5 fileMode = fileStats [ stat.ST_MODE ]

6 if stat.S_ISREG ( fileStats [ stat.ST_MODE ] ):

7 print 'Regular file.'

8 elif stat.S_ISDIR ( fileStats [ stat.ST_MODE ] ):

9 print 'Directory.'

10 elif stat.S_ISLNK ( fileStats [ stat.ST_MODE ] ):

11 print 'Shortcut.'

12 elif stat.S_ISSOCK ( fileStats [ stat.ST_MODE ] ):13 print 'Socket.'

14elif stat.S_ISFIFO ( fileStats [ stat.ST_MODE ] ):15 print 'Named pipe.'

16 elif stat.S_ISBLK ( fileStats [ stat.ST_MODE ] ):17 print 'Block special device.'

18elif stat.S_ISCHR ( fileStats [ stat.ST_MODE ] ):19 print 'Character special device.'另外，我們可以使用“os.path”來獲取基本信息：

1 import os.path

2

3 fileStats = 'test.txt'

4 if os.path.isdir ( fileStats ):

5 print 'Directory.'

6 elif os.path.isfile ( fileStats ):7 print 'File.'

8 elif os.path.islink ( fileStats ):

9 print 'Shortcut.'

10 elif os.path.ismount ( fileStats ):

11 print 'Mount point.'三、目錄

和普通文件一樣，關(guān)于目錄的操作也很容易掌握。首先，列出一個(gè)目錄的內(nèi)容：

1 import os

23 for fileName in os.listdir ( '/' ):

4 print fileName

正如你所見，這很簡(jiǎn)單，用三行代碼就可以完成。

創(chuàng)建目錄也很簡(jiǎn)單：

1 import os

2

3 os.mkdir ( 'testDirectory' )

刪除剛才創(chuàng)建的目錄：1 import os

2

3 os.rmdir ( 'testDirectory )

嗯，可以創(chuàng)建多級(jí)目錄：

1 import os

2

3 os.makedirs ( 'I/will/show/you/how/deep/the/rabbit/hole/goes' )

如果沒有在創(chuàng)建的文件夾中添加任何東西，就可以一次性將它們?nèi)縿h除(即，刪除所列的所有空文件夾)：

1 import os

2

3 os.removedirs ( 'I/will/show/you/how/deep/the/rabbit/hole/goes' )

當(dāng)需要對(duì)一個(gè)特定的文件類型進(jìn)行操作時(shí)，我們可以選擇“fnmatch”模塊。以下是顯示“.txt”文件的內(nèi)容和“.exe”文件的文件名：

1 import fnmatch

2 import os

3

4 for fileName in os.listdir ( '/' ):

5 if fnmatch.fnmath ( fileName, '*.txt' ):

6 print open ( fileName ).read()

7 elif fnmatch.fnmatch ( fileName, '*.exe' ):

8 print fileName

“*”字符可以表示任意長(zhǎng)度的字符。如果要匹配一個(gè)字符，則使用“？”符號(hào)：

1 import fnmatch

2 import os

3

4 for fileName in os.listdir ( '/' ):

5 if fnmatch.fnmatch ( fileName, '?.txt' ):

6 print 'Text file.'

“fnmatch”模塊支持正則表達(dá)式：

1 import fnmatch

2 import os

3 import re

4

5 filePattern = fnmatch.translate ( '*.txt' )

6 for fileName in os.listdir ( '/' ):

7 if re.match ( filePattern, fileName ):8 print 'Text file.'

若只需要匹配一種類型的文件，更好的辦法是使用“glob”模塊。該模塊的格式和“fnmatch”相似：

1 import glob

2

3 for fileName in glob.glob ( '*.txt' )：

4 print 'Text file.'

使用一定范圍的字符來匹配同樣可行，就像在正則表達(dá)式中使用一樣。假設(shè)你想要顯示擴(kuò)展名前只有一位數(shù)字的文件的文件名：

1 import glob

2

3 for fileName in glob.glob ( '[0-9].txt' )：

4 print filename

“glob”模塊利用“fnmatch”模塊來實(shí)現(xiàn)。

四、數(shù)據(jù)編組

使用前一節(jié)中介紹的模塊，可以實(shí)現(xiàn)在文件中對(duì)字符串的讀寫。

然而，有的時(shí)候，你可能需要傳遞其它類型的數(shù)據(jù)，如list、tuple、dictionary和其它對(duì)象。在Python中，你可以使用Pickling來完成。你可以使用Python標(biāo)準(zhǔn)庫中的“pickle”模塊完成數(shù)據(jù)編組。

下面，我們來編組一個(gè)包含字符串和數(shù)字的list：

1 import pickle

2

3 fileHandle = open ( 'pickleFile.txt'， 'w' )

4 testList = [ 'This'， 2， 'is'， 1， 'a'， 0， 'test.' ]

5 pickle.dump ( testList， fileHandle )

6 fileHandle.close()

拆分編組同樣不難：

1 import pickle

2

3 fileHandle = open ( 'pickleFile.txt' )

4 testList = pickle.load ( fileHandle )

5 fileHandle.close()

1 import pickle

2

3 fileHandle = open ( 'pickleFile.txt'， 'w' )

4 testList = [ 123， { 'Calories' ： 190 }， 'Mr. Anderson'， [ 1， 2， 7 ] ]

5 pickle.dump ( testList， fileHandle )

6 fileHandle.close()

1 import pickle

2

3 fileHandle = open ( 'pickleFile.txt' )

4 testList = pickle.load ( fileHandle )

5 fileHandle.close()

如上所述，使用Python的“pickle”模塊編組確實(shí)很簡(jiǎn)單。眾多對(duì)象可以通過它來存儲(chǔ)到文件中。如果可以的話，“cPickle”同樣勝任這個(gè)工作。它和“pickle”模塊一樣，但是速度更快：

1 import cPickle

2

3 fileHandle = open ( 'pickleFile.txt'， 'w' )

4 cPickle.dump ( 1776， fileHandle )

5 fileHandle.close()

五、創(chuàng)建“虛擬”文件

你用到的許多模塊包含需要文件對(duì)象作為參數(shù)的方法。但是，有時(shí)創(chuàng)建并使用一個(gè)真實(shí)的文件并讓人感到有些麻煩。所幸的是，在Python中，你可以使用“StringIO”模塊來創(chuàng)建文件并將其保存在內(nèi)存中：

1 import StringIO

2

3 fileHandle = StringIO.StringIO ( 'Let freedom ring' )

4 print fileHandle.read() # 'Let freedom ring.'

5 fileHandle.close()

cStringIO”模塊同樣有效。它的使用方法和“StringIO”一樣，但就像“cPickle”之于“pickle”，它速度更快：

1 import cStringIO

2

3 fileHandle = cStringIO.cStringIO ( 'To Kill a Mockingbird' )

4 print fileHandle.read() # 'To Kill a Mockingbid'

5 fileHandle.close()

結(jié)論

文件管理，是眾多編程語言的程序員在編寫應(yīng)用程序是經(jīng)常遇到的問題。幸好，和其它語言相比，Python使其出乎意料地容易。Python的標(biāo)準(zhǔn)庫中提供了許多相關(guān)的模塊幫助程序員解決這方面的問題，而它的面向?qū)ο蟮臋C(jī)制也簡(jiǎn)化了操作。

總結(jié)

以上是生活随笔為你收集整理的python的模块提供了许多文件管理方法_学会这几招，轻松掌握Python文件管理的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。