2022年 11月 3日

Python统计文件行数

问：统计项目下文件的总行数？

解：
        第一步：先用python统计单个文件行数；
        第二步：遍历文件夹累加全部文件行数；

当然，pycharm有插件直接可以统计行数：Statistic – IntelliJ IDEs Plugin | Marketplace

具体实现步骤：获取单个文件行数：

  1. # -*- coding: utf-8 -*-
  2.  
  3. file = r"文件路径"
  4.  
  5. # 第一种，直接打开文件读取行数（文件较小时）
  6. count = 1
  7. for count, line in enumerate(open(file, 'r', encoding='utf-8').readlines()):
  8.     count += 1
  9. print('行数:', count)
  10.  
  11. # 第二种，读取文件到缓存中，根据换行符进行计算
  12. count = 1
  13. fp = open(file, "r", encoding='utf-8')
  14. while 1:
  15.     buffer = fp.read(8*1024*1024)
  16.     if not buffer:
  17.         break
  18.     count += buffer.count('\n')
  19. print("行数:", count)

遍历文件夹进行读取：这里需要引入os模块，对文件夹进行遍历

  1. import os
  2.  
  3. total_line_num = 0  # 总行数
  4. total_file_num = 0  # 总文件数
  5. avg_file_num = 0    # 平均行数
  6. max_line_num = 0    # 最大行数
  7. min_line_num = 999999    # 最小行数
  8.  
  9. for base_path, folder_list, file_list in os.walk('C:/Users/xxx/xxx/xxx'):
  10.  
  11.     total_file_num += len(file_list)
  12.     # 遍历文件列表
  13.     for file_name in file_list:
  14.         # 文件路径
  15.         file_path = os.path.join(base_path, file_name)
  16.         # 获取文件后缀，根据需要根据后缀进行文件排除
  17.         file_ext = file_path.rsplit('.', maxsplit=1)
  18.         # 不是py文件,排除
  19.         if file_ext[1] != 'py':
  20.             continue
  21.         # py文件数加一
  22.         total_file_num += 1
  23.         count = 0
  24.         with open(file_path, 'rb') as f:
  25.             for line in f:
  26.                 # 根据需要是否去除空格
  27.                 line = line.strip()
  28.                 if not line:
  29.                     continue
  30.                 # 根据需要是否去除注释
  31.                 if line.startswith(b'#'):
  32.                     continue
  33.                 count += 1
  34.         total_line_num += count
  35.         max_line_num = count if count > max_line_num else max_line_num
  36.         min_line_num = count if count < min_line_num else min_line_num
  37. print('总行数：', total_line_num)
  38. print('总文件数：', total_file_num)
  39. print('最多行数：', max_line_num)
  40. print('最小总行数：', min_line_num)
  41. print('平均行数：', total_line_num / total_file_num)