【Python】9個必知的Python操作文件/文件夾方法

輸出當(dāng)前的路徑

我們可以通過Python當(dāng)中的OS庫來獲取當(dāng)前文件所在的位置

import?os
os.getcwd()

路徑的拼接

我們通過os.path.join()方法來實(shí)現(xiàn)

os.path.join('output',?'Book1.xlsx')

output

output\Book1.xlsx

確認(rèn)某文件夾或者是文件是否存在

我們先來看如何來確認(rèn)文件夾是否存在，同樣也是通過OS模塊

os.path.exists('directory_name')

同樣的道理，要是我們查看一下某個文件是否存在的時候，可以這么做

os.path.exists('path/file_name')

建立文件夾目錄

然后我們來看一下如何新建一個文件夾

os.mkdir("文件夾名稱")

當(dāng)然要是文件夾目錄已經(jīng)提前存在的情況下，上述的代碼自然會報錯，所以我們通常會先檢查一下是否已經(jīng)存在

if?not?os.path.exists('文件夾名稱'):
????os.mkdir('文件夾名稱')

當(dāng)然我們有時候需要在已建的文件夾下創(chuàng)建子文件夾，例如下圖

os.makedirs(os.path.join('test_dir',?'level_1',?'level_2',?'level_3'))

羅列出當(dāng)前目錄下所包含的所有文件

代碼如下

os.listdir('文件夾名稱')

list(glob(os.path.join('.',?'*.py')))

上面提到的glob模塊能夠快速查找我們想要的目錄和文件，它支持*、**、?、[]這四個通配符

移動文件

import?shutil

for?file?in?list(glob(os.path.join('.',?'*.csv'))):
????shutil.move(file,?'test_dir')

復(fù)制文件

shutil.copy(os.path.join('test_dir',?'data.csv'),?'output')

除此之外，我們還能夠?qū)φ迟N過去的文件重命名，代碼如下

shutil.copy(os.path.join('test_dir',?'data.csv'),
????????????os.path.join('output',?'data_2.csv'))

刪除文件

我們來看一下刪除文件該如何實(shí)現(xiàn)？os.remove()方法可以完成刪除文件的功能，

os.remove(os.path.join('output',?'data_2.csv'))

os.rmdir(os.path.join('test_dir',?'level_1',?'level_2',?'level_3'))

shutil.rmtree("test_delete")

或者是

shutil.rmtree(os.path.join("test_delete",?"test_1_delete"))

創(chuàng)建和解壓 壓縮包

談及到操作壓縮包相關(guān)的內(nèi)容，就不得不說zipobj這個模塊了

• 創(chuàng)建一個壓縮包

這里我們要用到的是zipfile模塊當(dāng)中的write()方法

file_lists?=?list(glob(os.path.join('.',?'*.xlsx')))

with?zipfile.ZipFile(r"我創(chuàng)建的壓縮包.zip",?"w")?as?zipobj:
????for?file?in?file_lists:
????????zipobj.write(file)

• 讀取壓縮包當(dāng)中的文件信息

通過zipfile模塊當(dāng)中的namelist()方法來實(shí)現(xiàn)

with?zipfile.ZipFile("我創(chuàng)建的壓縮包.zip",?"r")?as?zipobj:
????print(zipobj.namelist())

output

['Book1.xlsx',?'supermarkt_sales.xlsx']

可以看到輸出的就是上次我們打包的幾個excel文件

• 將壓縮包當(dāng)中的單個文件，解壓出來

通過zipfile模塊當(dāng)中的extract()方法來實(shí)現(xiàn)

dst?=?"output"

with?zipfile.ZipFile("我創(chuàng)建的壓縮包.zip",?"r")?as?zipobj:
????zipobj.extract("Book1.xlsx",dst)

上述代碼的用意就是將壓縮包當(dāng)中的“Book1.xlsx”文件解壓至“output”目錄文件夾下

• 將壓縮包中的所有文件，都解壓出來

通過zipfile模塊當(dāng)中的extractall()方法來實(shí)現(xiàn)，代碼如下

dst?=?"output"

with?zipfile.ZipFile("我創(chuàng)建的壓縮包.zip",?"r")?as?zipobj:
????zipobj.extractall(dst)