溫馨提示×

您好，登錄后才能下訂單哦！

密碼登錄×
登錄注冊(cè)×
其他方式登錄
點(diǎn)擊 登錄注冊(cè) 即表示同意《億速云用戶(hù)服務(wù)條款》

Python 對(duì)文件目錄操作的方法

發(fā)布時(shí)間：2020-07-11 10:21:33 來(lái)源：億速云 閱讀：199 作者：清晨 欄目：開(kāi)發(fā)技術(shù)

小編給大家分享一下Python 對(duì)文件目錄操作的方法，希望大家閱讀完這篇文章后大所收獲，下面讓我們一起去探討吧！

概述

I/O操作不僅包括屏幕輸入輸出，還包括文件的讀取與寫(xiě)入，Python提供了很多必要的方法和功能，進(jìn)行文件及文件夾的相關(guān)操作。本文主要通過(guò)兩個(gè)簡(jiǎn)單的小例子，簡(jiǎn)述Python在文件夾及文件的應(yīng)用，僅供學(xué)習(xí)分享使用，如有不足之處，還請(qǐng)指正。

涉及知識(shí)點(diǎn)

os模塊：os 模塊提供了非常豐富的方法用來(lái)處理文件和目錄。
open方法：open方法用于打開(kāi)一個(gè)文件，用于讀取和寫(xiě)入。

實(shí)例1：獲取指定目錄下所有的文件大小，并找出最大文件及最小文件

分解步驟：

遍歷文件夾下所有的子文件及子文件夾（需要遞歸），并計(jì)算每一個(gè)文件的大小
計(jì)算所有文件的大小總和
找出最大文件及最小文件

核心代碼

定義一個(gè)方法get_file_size，獲取單個(gè)文件的大小，單位有KB和MB兩種。關(guān)鍵點(diǎn)如下所示：

  1. os.path.getsize 用于獲取指定文件的大小，單位是Byte。
  2. round為四舍五入函數(shù)，保留指定位數(shù)的小數(shù)。
def get_file_size(file_path, KB=False, MB=False):
  """獲取文件大小"""
  size = os.path.getsize(file_path)
  if KB:
    size = round(size / 1024, 2)
  elif MB:
    size = round(size / 1024 * 1024, 2)
  else:
    size = size
  return size

定義一個(gè)方法list_files，遍歷指定文件目錄，并存入字典當(dāng)中。關(guān)鍵點(diǎn)如下所示：

  • os.path.isfile 用于判斷給定的路徑是文件還是文件夾。
  • os.listdir 用于獲取指定目錄下所有的文件及文件夾，返回一個(gè)列表，但是只是當(dāng)前文件夾的名稱(chēng)，并不是全路徑。
  • os.path.join 用于拼接兩個(gè)路徑
def list_files(root_dir):
  """遍歷文件"""
  if os.path.isfile(root_dir): # 如果是文件
    size = get_file_size(root_dir, KB=True)
    file_dict[root_dir] = size
  else:
    # 如果是文件夾，則遍歷
    for f in os.listdir(root_dir):
      # 拼接路徑
      file_path = os.path.join(root_dir, f)
      if os.path.isfile(file_path):
        # 如果是一個(gè)文件
        size = get_file_size(file_path, KB=True)
        file_dict[file_path] = size
      else:
        list_files(file_path)

計(jì)算總大小和最大文件及最小文件，如下所示：

通過(guò)比較字典value的大小，返回對(duì)應(yīng)的key的名稱(chēng)。關(guān)鍵點(diǎn)如下所示：

  1. max_file = max(file_dict, key=lambda x: file_dict[x])
  2. min_file = min(file_dict, key=lambda x: file_dict[x])
if __name__ == '__main__':
  list_files(root_dir)
  # print( len(file_dict))
  # 計(jì)算文件目錄大小
  total_size = 0
  # 遍歷字典的key
  for file in file_dict:
    total_size += file_dict[file]

  print('total size is : %.2f' % total_size)
  # 找最大最小文件
  max_file = max(file_dict, key=lambda x: file_dict[x])
  min_file = min(file_dict, key=lambda x: file_dict[x])
  print('max file is : ', max_file, '\n file size is :', file_dict[max_file])
  print('min file is : ', min_file, '\n file size is :', file_dict[min_file])

實(shí)例2：將兩個(gè)文本文件中的內(nèi)容進(jìn)行合并，并保存到文件中

兩個(gè)文件內(nèi)容，如下圖所示：

Python 對(duì)文件目錄操作的方法

分解步驟：

  1. 讀取兩個(gè)文件中的內(nèi)容并進(jìn)行解析出key和value，存入字典當(dāng)中（兩個(gè)字典，分別存儲(chǔ)兩個(gè)文件的內(nèi)容）。
  2. 遍歷第1個(gè)字典，并查找第2個(gè)字典當(dāng)中相同key的值，如不存在，則只顯示第1個(gè)內(nèi)容，如存在，則進(jìn)行合并。
  3. 遍歷第2個(gè)字典，并查找不在第1個(gè)字典當(dāng)中的key的值，進(jìn)行累加。
  4. 將拼接后的列表存入文件當(dāng)中。
     

核心代碼

定義一個(gè)函數(shù)read_book，用于讀取兩個(gè)文件的內(nèi)容。關(guān)鍵點(diǎn)如下所示：

  1. open函數(shù)用于打開(kāi)一個(gè)文件，文件編碼為UTF-8。
  2. readlines用于讀取所有的行，并返回一個(gè)列表。
  3. split用于分割字符串為數(shù)組。
def read_book():
  """讀取內(nèi)容"""
  # 讀取一個(gè)文件
  file1 = open('book1.txt', 'r', encoding='UTF-8')
  lines1 = file1.readlines()
  file1.close()
  for line in lines1:
    line = line.strip() # 去空白
    content = line.split(',')
    book1[content[0]] = content[1]

  # 另一種方式，讀取另一個(gè)文件,不需要close，會(huì)自動(dòng)關(guān)閉
  with open('book2.txt', 'r', encoding='UTF-8') as file2:
    lines2 = file2.readlines()
  for line in lines2:
    line = line.strip() # 去空白
    content = line.split(',')
    book2[content[0]] = content[1]

定義一個(gè)函數(shù)，用于合并內(nèi)容，并保存。關(guān)鍵點(diǎn)如下所示：

  1. append 用于為數(shù)組添加新元素。
  2. dict.keys函數(shù) 用于返回所有的key。
  3. join函數(shù)用于將數(shù)組轉(zhuǎn)換成字符串，并以對(duì)應(yīng)字符分割。
  4. writelines 用于寫(xiě)入所有的行到文件。
  5. with語(yǔ)法，當(dāng)執(zhí)行結(jié)束時(shí)，自動(dòng)close，并釋放資源。
def merge_book():
  """合并內(nèi)容"""
  lines = [] # 定義一個(gè)空列表
  header = '姓名\t 電話\t 文本\n'
  lines.append(header)
  # 遍歷第一個(gè)字典
  for key in book1:
    line = ''
    if key in book2.keys():
      line = line + '\t'.join([key, book1[key], book2[key]])
      line += '\n'
    else:
      line = line + '\t'.join([key, book1[key], ' *****'])
      line += '\n'
    lines.append(line)
  # 遍歷第2個(gè)，將不包含在第1個(gè)里面的寫(xiě)入
  for key in book2:
    line = ''
    if key not in book1.keys():
      line = line + '\t'.join([key, ' *****', book2[key]])
      line += '\n'
      lines.append(line)
  # 寫(xiě)入book3
  with open('book3.txt', 'w', encoding='UTF-8') as f:
    f.writelines(lines)

整體調(diào)用，如下所示：

if __name__ == '__main__':
  # 讀取內(nèi)容
  read_book()
  # 合并內(nèi)容
  merge_book()

  # print(book1)
  # print(book2)

最后拼接后生成的文件，如下所示：

Python 對(duì)文件目錄操作的方法

看完了這篇文章，相信你對(duì)Python 對(duì)文件目錄操作的方法有了一定的了解，想了解更多相關(guān)知識(shí)，歡迎關(guān)注億速云行業(yè)資訊頻道，感謝各位的閱讀！

向AI問(wèn)一下細(xì)節(jié)

免責(zé)聲明：本站發(fā)布的內(nèi)容（圖片、視頻和文字）以原創(chuàng)、轉(zhuǎn)載和分享為主，文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng)，如果涉及侵權(quán)請(qǐng)聯(lián)系站長(zhǎng)郵箱：is@yisu.com進(jìn)行舉報(bào)，并提供相關(guān)證據(jù)，一經(jīng)查實(shí)，將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI
助
手