跳至內容

Python/文件

維基教科書，自由的教學讀本
上一節：模塊 返回目錄 下一節：文本

文件I/O

[編輯]

讀整個文件： 

inputFileText = open("testit.txt", "r").read()
print(inputFileText)

參數值"r"表示文件只讀模式打開。

從文件中讀入指定數量的字節： 

inputFileText = open("testit.txt", "r").read(123)
print(inputFileText)

打開文件後，用seek()重定位文件當前位置，用tell()獲取文件當前位置。如下例： 

>>> f=open("/proc/cpuinfo","r")
>>> f.tell()
0L
>>> f.read(10)
'processor\t'
>>> f.read(10)
': 0\nvendor'
>>> f.tell()
20L
>>> f.seek(10)
>>> f.tell()
10L
>>> f.read(10)
': 0\nvendor'
>>> f.close()
>>> f
<closed file '/proc/cpuinfo', mode 'r' at 0xb7d79770>

close()函數用於關閉一個打開的文件。

一次讀一行： 

for line in open("testit.txt", "r"):
    print(line)

readlines()函數將返回文件的所有行，作為字符串組成的列表。每個字符串以換行符結尾。

readline()函數讀取當前行，作為字符串返回。字符串以換行符結尾。

寫文件時，需要在open()用參數"w"： 

outputFileText = "Here's some text to save in a file"
open("testit.txt", "w").write(outputFileText)

追加文件時，需要在open()用參數"a"： 

outputFileText = "Here's some text to add to the existing file."
open("testit.txt", "a").write(outputFileText)

從Python 2.5開始，可以用關鍵字with保證文件句柄被釋放、異常安全： 

with open("input.txt") as file1:
  data = file1.read()
  # process the data

或者： 

with open("input.txt") as file1:
  for line in file1:
    print(line)

關於關鍵字with，參見Context Managers。

外部連結：

對文件的相關測試

[編輯]

確定路徑是否存在： 

import os
os.path.exists('<path string>')

對於Microsoft Windows™，可寫為： 

import os
os.path.exists('C:\\windows\\example\\path')

或寫成"raw"字符串（r）： 

import os
os.path.exists(r'C:\windows\example\path')

os.path還包括一些有用函數： 

>>> import os
>>> os.path.isfile("/")
False
>>> os.path.isfile("/proc/cpuinfo")
True
>>> os.path.isdir("/")
True
>>> os.path.isdir("/proc/cpuinfo")
False
>>> os.path.ismount("/")
True
>>> os.path.islink("/")
False
>>> os.path.islink("/vmlinuz")
True
>>> os.path.realpath("/vmlinuz")
'/boot/vmlinuz-2.6.24-21-generic'

文件常見操作

[編輯]

copy或move文件，使用庫shutil 

import shutil
shutil.move("originallocation.txt","newlocation.txt")
shutil.copy("original.txt","copy.txt")

對一個目錄的遞歸複製用copytree()，對一個目錄的遞歸move用rmtree()： 

import shutil
shutil.copytree("dir1","dir2")
shutil.rmtree("dir1")

刪除一個文件，使用os庫中的remove()函數： 

import os
os.remove("file.txt")

查找文件

[編輯]

可以用glob查找文件： 

glob.glob('*.txt') # Finds files in the currect directory ending in dot txt 
glob.glob('*\\*.txt') # Finds files in any of the direct subdirectories
                      # of the currect directory ending in dot txt 
glob.glob('C:\\Windows\\*.exe')
for fileName in glob.glob('C:\\Windows\\*.exe'):
  print(fileName)
glob.glob('C:\\Windows\\**\\*.exe', recursive=True) # Py 3.5: ** allows recursive nesting

glob通配符：

通配符 功能
* 匹配0或多個字符
** 匹配所有文件、目錄、子目錄和子目錄里的文件（3.5版本新增）
? 匹配1個字符，與正則表達式里的?不同
[exp] 匹配指定範圍內的字符，如：[1-9]匹配1至9範圍內的字符
[!exp] 匹配不在指定範圍內的字符

目錄的內容可以用listdir函數列出： 

filesAndDirectories=os.listdir('.')
for item in filesAndDirectories:
  if os.path.isfile(item) and item.endswith('.txt'):
    print "Text file: " + item
  if os.path.isdir(item):
    print "Directory: " + item

得到一個目錄下所有項目，包括在子目錄下的： 

for root, directories, files in os.walk('/user/Joe Hoe'):
  print "Root: " + root                          # e.g. /user/Joe Hoe/Docs
  for dir1 in directories:
    print "Dir.: " + dir1                        # e.g. Fin
    print "Dir. 2: " + os.path.join(root, dir1)  # e.g. /user/Joe Hoe/Docs/Fin
  for file1 in files:
    print "File: " + file1                       # e.g. MyFile.txt
    print "File 2: " + os.path.join(root, file1) # e.g. /user/Joe Hoe/Docs/MyFile.txt

得到一個目錄下的所有.txt，包括子目錄，使用list comprehension： 

files = [os.path.join(r, f) for r, d, fs in os.walk(".") for f in fs
         if f.endswith(".txt")]
# As iterator
files = (os.path.join(r, f) for r, d, fs in os.walk(".") for f in fs
         if f.endswith(".txt"))

外部連結：

當前目錄

[編輯]

得到當前工作目錄： 

os.getcwd()

改變當前工作目錄： 

os.chdir('C:\\')

外部連結

[編輯]
上一節：模塊 返回目錄 下一節：文本
取自「https://zh.wikibooks.org/w/index.php?title=Python/文件&oldid=133856」