1. 程式人生 > 其它 >Python檔案及目錄處理方法

Python檔案及目錄處理方法

Python可以用於處理文字檔案和二進位制檔案,比如建立檔案、讀寫檔案等操作。本文介紹Python處理目錄以及檔案的相關方法。

目錄

下面先來介紹python目錄處理相關方法。

目錄操作

1. 獲取當前程式碼路徑

test_folder.py

import os
import sys

print(__file__)
print(sys.argv[0])
print(os.path.realpath(__file__))
print(os.path.abspath(sys.argv[0]))

Out:

D:/ProgramWorkspace/PythonNotes/03-File-Handling/test_folder.py
D:/ProgramWorkspace/PythonNotes/03-File-Handling/test_folder.py
D:\ProgramWorkspace\PythonNotes\03-File-Handling\test_folder.py
D:\ProgramWorkspace\PythonNotes\03-File-Handling\test_folder.py

2. 獲取當前檔案__file__的所在目錄

print(os.getcwd())
print(os.path.dirname(os.path.realpath(__file__)))
print(os.path.split(os.path.realpath(__file__))[0])
path = os.path.dirname(os.path.realpath(__file__))

Out:

D:\ProgramWorkspace\PythonNotes\03-File-Handling
D:\ProgramWorkspace\PythonNotes\03-File-Handling
D:\ProgramWorkspace\PythonNotes\03-File-Handling

3. 獲取當前檔名名稱

print(os.path.basename(sys.argv[0])) # 當前檔名名稱
print(os.path.basename(__file__))

Out:

test_folder.py
test_folder.py

4. 拼接路徑

path = os.path.dirname(os.path.realpath(__file__))
filename = os.path.basename(__file__)
abspath = os.path.join(path, filename)
print(abspath)

Out:

D:\ProgramWorkspace\PythonNotes\03-File-Handling\test_folder.py

5. 建立目錄

判斷目錄是否存在:

os.path.exists(path)

建立目錄

if not os.path.exists(path): 
    print(f"建立資料夾: {path}")
    os.makedirs(path)

檔案操作

1. 建立文字檔案

text = "Hello World!"
newfilepath = os.path.join(path, "newfile.txt")
file = open(newfilepath, 'w')
file.write(text)  # 寫入內容資訊
file.close()

2. 判斷檔案是否存在

print(os.path.isfile(path)) 
print(os.path.isfile(newfilepath))
print(os.path.exists(newfilepath))

Out:

False
True
True

os.path.isfile 用於判斷是否為檔案且是否存在,os.path.exists 也可以用於判斷檔案是否存在,但還是建議使用os.path.isfile 判斷檔案,os.path.exists 判斷目錄是否存在。比如,某個檔案為newfile,使用這兩個方法都會返回True,無法判斷到底是檔案還是目錄。

3. 判斷檔案屬性

print(os.access(newfilepath,os.F_OK)) # 檔案是否存在
print(os.access(newfilepath,os.R_OK)) # 檔案是否可讀
print(os.access(newfilepath,os.W_OK)) # 檔案是否可以寫入
print(os.access(newfilepath,os.X_OK)) # 檔案是否有執行許可權

Out:

True
True
True
True

os.access(newfilepath,os.F_OK) 也可以用於判斷檔案是否存在。

4. 開啟檔案

開啟文字檔案或者二進位制檔案可以使用 open() 方法:

f = open(filename, mode)

幾種檔案開啟模式:

  • b:二進位制模式
  • t:文字模式(預設)
  • r: 開啟存在的檔案,讀操作(預設)。
  • w: 開啟檔案,寫操作,先前檔案中的內容會被刪除。如果檔案不存在會自動建立。
  • a: 開啟檔案,追加操作,不會刪除先前檔案中的內容。如果檔案不存在會自動建立。
  • x:建立新檔案,寫操作,使用此模式開啟存在的檔案會丟擲異常。
  • r+: 讀、寫操作,不會刪除先前檔案中的內容,但是會覆蓋內容。
  • w+: 寫、讀操作,會刪除先前檔案中的內容。
  • a+: 追加、讀操作,不會刪除和覆蓋先前檔案中的內容。
  • x+ :建立新檔案,讀寫操作。

open()方法的預設模式為 rt 模式,也就是讀文字檔案。

另外需要注意filename的寫法,比如檔案路徑是:D:\test.txt ,其中\t 可能會被轉義,需要自前面加一個 r

f = open(r"D:\test.txt", "w")

5. 寫檔案

讀寫檔案都需要先開啟檔案,返回一個檔案物件,然後對檔案物件進行讀寫操作。寫檔案需要設定寫許可權,比如 ww+a 模式。

寫檔案主要包括兩種方法:

  • file.write(str):寫入字串
  • file.writelines(list):寫入字串列表,用於同時插入多個字串。

舉個栗子:

file = open("newfile.txt", 'w')
text1 = "Hello World!\n你好,世界!\r"
file.write(text1)  # 寫入內容資訊

text2 = ["To the time to life, \n", "rather than to life in time.\r"]
file.writelines(text2)

file.close()

w 模式會刪除先前檔案中的內容,如果不想刪除,需要直接追加到後面,可以使用aa+ 模式:

file = open("newfile.txt", 'a')

6. 讀檔案

常見的讀取檔案方法有以下幾種:

  • in操作符
  • read():讀取所有資料,返回一個字串。
  • readline():讀取第一行
  • readlines():讀取所有行,每行儲存為列表的一個元素。
# 開啟並讀取檔案
file = open("newfile.txt", 'r')

for line in file:
	print(line)
print()

file.seek(0, 0)
print(file.read(5))  # 
print()

file.seek(0, 0)
print(file.readline(12))
print()

file.seek(0, 0)
print(file.readlines())
print()

file.close()

執行結果:

Hello World!

你好,世界!

To the time to life, 

rather than to life in time.


Hello

Hello World!

['Hello World!\n', '你好,世界!\n', 'To the time to life, \n', 'rather than to life in time.\n']

讀取file物件的所有內容後,文字的游標會移動到最後,再次讀取file需要將游標移到前面,使用 file.seek(0, 0) 方法可以將游標移到前面。還有一種解決方案是將讀取的內容存一個在區域性變數中。

7. with語句

with語句可用於異常處理,可以確保資源的適當獲取及自動釋放。使用with語句後就不需要呼叫file.close() 語句了,它會自動釋放。

text1 = "Hello World!\n你好,世界!\r"
text2 = ["To the time to life, \n", "rather than to life in time.\r"]
# 寫
with open("newfile.txt", "w") as file:
    file.write(text1)
    file.writelines(text2)

# 讀
with open("newfile.txt", "r+") as file:
	print(file.read())

with語句對處理大檔案非常有用,比如10G大小的檔案, with語句會進行上下文管理。

--THE END--