10行Python代碼助你整理雜亂無章的文件

更新時間：2023年02月16日 10:44:40 作者：派森醬

這篇文章主要和大家分享一個簡單的小程序，只需要10行Python代碼，就能助你整理雜亂無章的文件，感興趣的小伙伴可以跟隨小編一起了解一下

按照文件格式整理

具體來說就是不同的文件后綴名稱放置到不同的文件夾當(dāng)中

今天我們只需要使用 os 和 shutil 兩個庫就足夠了。os 是 Python 內(nèi)置模塊，用于操作系統(tǒng)的相關(guān)功能，shutil 是用來進(jìn)行文件移動等操作的。

首先我們先來看幾個獲取文件后綴的方法

獲取文件后綴

方法1：

直接通過 os 庫實現(xiàn)

filename?=?os.path.splitext('1.jpg')

Output:

('1', '.jpg')

得到一個包含文件名和后綴名的元組，此時再想獲得文件后綴就很容易了吧~

方法2：

可以直接使用字符串方法 endwith，簡單明了

file?=?"1.jpg".endswith(".jpg")

如果結(jié)尾字符是 “.jpg” 的話，那么該方法的返回值則為 True，以此我們可以判斷文件的后綴名

方法3：

使用字符串的 split 方法，通過切分來獲取后綴名稱

file1?=?"1.jpg".split('.')

這樣可以得到一個列表，第二個元素就是后綴名

接下來我們就可以創(chuàng)建文件夾和拷貝文件了，分別用 os.mkdir 和 shutil.copy 來操作

代碼如下：

import?os
import?shutil


files?=?os.listdir(".")

for?f?in?files:
????if?os.path.isfile(f)?and?not?f.startswith('.'):
????????folder_name?=?os.path.splitext(f)[-1]
????????folder_name?=?folder_name.split(".")[-1]
????????print(folder_name)
????????if?not?os.path.exists(folder_name):
????????????os.mkdir(folder_name)
????????????shutil.copy(f,?folder_name)
????????else:
????????????shutil.copy(f,?folder_name)

通過上面的代碼，我們就可以在當(dāng)前目錄下順利地整理文件了

按照文件大小整理

當(dāng)然有的時候，我們同樣希望把大小類似的文件整理到一個文件夾當(dāng)中，可以通過 os 庫的 getsize 函數(shù)來獲取

print(os.path.getsize("1.jpg"))

這樣就可以得到該文件的大小，單位是 KB

完整代碼如下：

import?os
import?shutil


files?=?os.listdir(".")

for?f?in?files:
????if?os.path.isfile(f)?and?not?f.startswith('.'):
????????size?=?os.path.getsize(f)
????????if?size?<?1000:
????????????folder_name?=?"Under_one_MB"
????????????if?not?os.path.exists(folder_name):
????????????????os.mkdir(folder_name)
????????????????shutil.copy(f,?folder_name)
????????????else:
????????????????shutil.copy(f,?folder_name)
????????elif?10000?<=?size?<?2000:
????????????folder_name?=?"Under_two_MB"
????????????if?not?os.path.exists(folder_name):
????????????????os.mkdir(folder_name)
????????????????shutil.copy(f,?folder_name)
????????????else:
????????????????shutil.copy(f,?folder_name)

好了，以上就是今天分享的文件整理小工具，還是蠻實用的，快來嘗試一下吧！

到此這篇關(guān)于10行Python代碼助你整理雜亂無章的文件的文章就介紹到這了,更多相關(guān)Python整理文件內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: