概述
I/O操作不仅包括屏幕输入输出,还包括文件的读取与写入,Python提供了很多必要的方法和功能,进行文件及文件夹的相关操作。本文主要通过两个简单的小例子,简述Python在文件夹及文件的应用,仅供学习分享使用,如有不足之处,还请指正。
涉及知识点
实例1:获取指定目录下所有的文件大小,并找出最大文件及最小文件
分解步骤:
核心代码
定义一个方法get_file_size,获取单个文件的大小,单位有KB和MB两种。关键点如下所示:
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,1)"> 8 else: 9 size = size 10 return size
定义一个方法list_files,遍历指定文件目录,并存入字典当中。关键点如下所示:
- os.path.isfile 用于判断给定的路径是文件还是文件夹。
- os.listdir 用于获取指定目录下所有的文件及文件夹,返回一个列表,但是只是当前文件夹的名称,并不是全路径。
- os.path.join 用于拼接两个路径
def list_files(root_dir): 遍历文件 3 if os.path.isfile(root_dir): # 如果是文件 4 size = get_file_size(root_dir,KB=True) 5 file_dict[root_dir] = 7 如果是文件夹,则遍历 8 for f in os.listdir(root_dir): 9 拼接路径 10 file_path = os.path.join(root_dir,f) 11 os.path.isfile(file_path): 12 如果是一个文件 13 size = get_file_size(file_path,1)">14 file_dict[file_path] =15 16 list_files(file_path)
通过比较字典value的大小,返回对应的key的名称。关键点如下所示:
- max_file = max(file_dict,key=lambda x: file_dict[x])
- min_file = min(file_dict,key=lambda x: file_dict[x])
if __name__ == '__main__' 2 list_files(root_dir) print( len(file_dict)) 计算文件目录大小 5 total_size = 0 遍历字典的key 7 for file file_dict: 8 total_size += file_dict[file] 9 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,1)">14 max file is : ',max_file,\n file size is :,file_dict[max_file]) 15 min file is : 实例2:将两个文本文件中的内容进行合并,并保存到文件中
分解步骤:
- 读取两个文件中的内容并进行解析出key和value,存入字典当中(两个字典,分别存储两个文件的内容)。
- 遍历第1个字典,并查找第2个字典当中相同key的值,如不存在,则只显示第1个内容,如存在,则进行合并。
- 遍历第2个字典,并查找不在第1个字典当中的key的值,进行累加。
- 将拼接后的列表存入文件当中。
核心代码
定义一个函数read_book,用于读取两个文件的内容。关键点如下所示:
read_book(): 读取内容 读取一个文件 4 file1 = open(book1.txtrUTF-8 5 lines1 = file1.readlines() 6 file1.close() for line lines1: 8 line = line.strip() 去空白 9 content = line.split(10 book1[content[0]] = content[1] 11 12 另一种方式,读取另一个文件,不需要close,会自动关闭 13 with open(book2.txt) as file2: 14 lines2 = file2.readlines() lines2: 16 line = line.strip() 17 content = line.split(18 book2[content[0]] = content[1]
- append 用于为数组添加新元素。
- dict.keys函数 用于返回所有的key。
- join函数用于将数组转换成字符串,并以对应字符分割。
- writelines 用于写入所有的行到文件。
- with语法,当执行结束时,自动close,并释放资源。
merge_book(): 合并内容 3 lines = [] 定义一个空列表 4 header = 姓名\t 电话\t 文本\n' 5 lines.append(header) 遍历第一个字典 for key book1: 8 line = '' 9 if key book2.keys(): 10 line = line + \t.join([key,book1[key],book2[key]]) 11 line += \n12 13 line = line + '.join([key,1)"> *****]) 14 line += 15 lines.append(line) 16 遍历第2个,将不包含在第1个里面的写入 17 book2: 18 line = 19 not book1.keys(): 20 line = line + 21 line += 22 lines.append(line) 23 写入book3 24 with open(book3.txtw) as f: 25 f.writelines(lines)整体调用,如下所示:
通过以上两个例子,可以大致了解文件及目录操作的一些方法及步骤。
备注
西江月·夜行黄沙道中
[宋]辛弃疾
明月别枝惊鹊,清风半夜鸣蝉。稻花香里说丰年,听取蛙声一片。
七八个星天外,两三点雨山前。旧时茅店社林边,路转溪桥忽见。