Python基礎之檔案目錄操作

概述

I/O操作不僅包括螢屏輸入輸出，還包括檔案的讀取與寫入，Python提供了很多必要的方法和功能，進行檔案及檔案夾的相關操作，本文主要通過兩個簡單的小例子，簡述Python在檔案夾及檔案的應用，僅供學習分享使用，如有不足之處，還請指正，

涉及知識點

1. os模塊：os 模塊提供了非常豐富的方法用來處理檔案和目錄，
2. open方法：open方法用于打開一個檔案，用于讀取和寫入，

實體1：獲取指定目錄下所有的檔案大小，并找出最大檔案及最小檔案

分解步驟：

1. 遍歷檔案夾下所有的子檔案及子檔案夾（需要遞回），并計算每一個檔案的大小
2. 計算所有檔案的大小總和
3. 找出最大檔案及最小檔案

核心代碼

定義一個方法get_file_size，獲取單個檔案的大小，單位有KB和MB兩種，關鍵點如下所示：

1. os.path.getsize 用于獲取指定檔案的大小，單位是Byte，
2. round為四舍五入函式，保留指定位數的小數，

 1 def get_file_size(file_path, KB=False, MB=False):
 2     """獲取檔案大小"""
 3     size = os.path.getsize(file_path)
 4     if KB:
 5         size = round(size / 1024, 2)
 6     elif MB:
 7         size = round(size / 1024 * 1024, 2)
 8     else:
 9         size = size
10     return size

定義一個方法list_files，遍歷指定檔案目錄，并存入字典當中，關鍵點如下所示：

1. os.path.isfile 用于判斷給定的路徑是檔案還是檔案夾，
2. os.listdir 用于獲取指定目錄下所有的檔案及檔案夾，回傳一個串列，但是只是當前檔案夾的名稱，并不是全路徑，
3. os.path.join 用于拼接兩個路徑

 1 def list_files(root_dir):
 2     """遍歷檔案"""
 3     if os.path.isfile(root_dir):  # 如果是檔案
 4         size = get_file_size(root_dir, KB=True)
 5         file_dict[root_dir] = size
 6     else:
 7         # 如果是檔案夾，則遍歷
 8         for f in os.listdir(root_dir):
 9             # 拼接路徑
10             file_path = os.path.join(root_dir, f)
11             if os.path.isfile(file_path):
12                 # 如果是一個檔案
13                 size = get_file_size(file_path, KB=True)
14                 file_dict[file_path] = size
15             else:
16                 list_files(file_path)

計算總大小和最大檔案及最小檔案，如下所示：

通過比較字典value的大小，回傳對應的key的名稱，關鍵點如下所示：

1. max_file = max(file_dict, key=lambda x: file_dict[x])
2. min_file = min(file_dict, key=lambda x: file_dict[x])

 1 if __name__ == '__main__':
 2     list_files(root_dir)
 3     # print( len(file_dict))
 4     # 計算檔案目錄大小
 5     total_size = 0
 6     # 遍歷字典的key
 7     for file in file_dict:
 8         total_size += file_dict[file]
 9 
10     print('total size is : %.2f' % total_size)
11     # 找最大最小檔案
12     max_file = max(file_dict, key=lambda x: file_dict[x])
13     min_file = min(file_dict, key=lambda x: file_dict[x])
14     print('max file is : ', max_file, '\n file size is :', file_dict[max_file])
15     print('min file is : ', min_file, '\n file size is :', file_dict[min_file])

實體2：將兩個文本檔案中的內容進行合并，并保存到檔案中

兩個檔案內容，如下圖所示：

分解步驟：

1. 讀取兩個檔案中的內容并進行決議出key和value，存入字典當中（兩個字典，分別存盤兩個檔案的內容），
2. 遍歷第1個字典，并查找第2個字典當中相同key的值，如不存在，則只顯示第1個內容，如存在，則進行合并，
3. 遍歷第2個字典，并查找不在第1個字典當中的key的值，進行累加，
4. 將拼接后的串列存入檔案當中，

核心代碼

定義一個函式read_book，用于讀取兩個檔案的內容，關鍵點如下所示：

1. open函式用于打開一個檔案，檔案編碼為UTF-8，
2. readlines用于讀取所有的行，并回傳一個串列，
3. split用于分割字串為陣列，

 1 def read_book():
 2     """讀取內容"""
 3     # 讀取一個檔案
 4     file1 = open('book1.txt', 'r', encoding='UTF-8')
 5     lines1 = file1.readlines()
 6     file1.close()
 7     for line in lines1:
 8         line = line.strip()  # 去空白
 9         content = line.split(',')
10         book1[content[0]] = content[1]
11 
12     # 另一種方式，讀取另一個檔案,不需要close，會自動關閉
13     with open('book2.txt', 'r', encoding='UTF-8') as file2:
14         lines2 = file2.readlines()
15     for line in lines2:
16         line = line.strip()  # 去空白
17         content = line.split(',')
18         book2[content[0]] = content[1]

定義一個函式，用于合并內容，并保存，關鍵點如下所示：

1. append 用于為陣列添加新元素，
2. dict.keys函式 用于回傳所有的key，
3. join函式用于將陣列轉換成字串，并以對應字符分割，
4. writelines 用于寫入所有的行到檔案，
5. with語法，當執行結束時，自動close，并釋放資源，

 1 def merge_book():
 2     """合并內容"""
 3     lines = []  # 定義一個空串列
 4     header = '姓名\t 電話\t 文本\n'
 5     lines.append(header)
 6     # 遍歷第一個字典
 7     for key in book1:
 8         line = ''
 9         if key in book2.keys():
10             line = line + '\t'.join([key, book1[key], book2[key]])
11             line += '\n'
12         else:
13             line = line + '\t'.join([key, book1[key], ' *****'])
14             line += '\n'
15         lines.append(line)
16     # 遍歷第2個，將不包含在第1個里面的寫入
17     for key in book2:
18         line = ''
19         if key not in book1.keys():
20             line = line + '\t'.join([key, ' *****', book2[key]])
21             line += '\n'
22             lines.append(line)
23     # 寫入book3
24     with open('book3.txt', 'w', encoding='UTF-8') as f:
25         f.writelines(lines)

整體呼叫，如下所示：

1 if __name__ == '__main__':
2     # 讀取內容
3     read_book()
4     # 合并內容
5     merge_book()
6 
7     # print(book1)
8     # print(book2)

最后拼接后生成的檔案，如下所示：

通過以上兩個例子，可以大致了解檔案及目錄操作的一些方法及步驟，

備注

西江月·夜行黃沙道中

[宋]辛棄疾

明月別枝驚鵲，清風半夜鳴蟬，稻花香里說豐年，聽取蛙聲一片，
七八個星天外，兩三點雨山前，舊時茅店社林邊，路轉溪橋忽見，

標籤：Python

上一篇：pandas學習之思維導圖

下一篇：odoo命名規范及常用函式