文件操作和OS模塊的簡單操作

文件的作用

大家應該聽說過一句話：“好記性不如爛筆頭”。

不僅人的大腦會遺忘事情，計算機也會如此，比如一個程序在運行過程中用了九牛二虎之力終於計算出了結果，試想一下如果不把這些數據存放起來，相比重啟電腦之后，“哭都沒地方哭了”

可見，在把數據存儲起來有做么大的價值

使用文件的目的：

　　就是把一些存儲存放起來，可以讓程序下一次執行的時候直接使用，而不必重新制作一份，省時省力

 

 

文件的打開與關閉

想一想：

如果想用word編寫一份簡歷，應該有哪些流程呢？

1. 打開word軟件，新建一個word文件
2. 寫入個人簡歷信息
3. 保存文件
4. 關閉word軟件

同樣，在操作文件的整體過程與使用word編寫一份簡歷的過程是很相似的

　　　　

1. 打開文件，或者新建立一個文件
2. 讀/寫數據
3. 關閉文件

 

打開文件

在python，使用open函數，可以打開一個已經存在的文件，或者創建一個新文件

open(文件名，訪問模式)

示例如下：

    f = open('test.txt', 'w') 

說明:

訪問模式	說明
r	以只讀方式打開文件。文件的指針將會放在文件的開頭。這是默認模式。
w	打開一個文件只用於寫入。如果該文件已存在則將其覆蓋。如果該文件不存在，創建新文件。
a	打開一個文件用於追加。如果該文件已存在，文件指針將會放在文件的結尾。也就是說，新的內容將會被寫入到已有內容之后。如果該文件不存在，創建新文件進行寫入。
rb	以二進制格式打開一個文件用於只讀。文件指針將會放在文件的開頭。這是默認模式。
wb	以二進制格式打開一個文件只用於寫入。如果該文件已存在則將其覆蓋。如果該文件不存在，創建新文件。
ab	以二進制格式打開一個文件用於追加。如果該文件已存在，文件指針將會放在文件的結尾。也就是說，新的內容將會被寫入到已有內容之后。如果該文件不存在，創建新文件進行寫入。
r+	打開一個文件用於讀寫。文件指針將會放在文件的開頭。
w+	打開一個文件用於讀寫。如果該文件已存在則將其覆蓋。如果該文件不存在，創建新文件。
a+	打開一個文件用於讀寫。如果該文件已存在，文件指針將會放在文件的結尾。文件打開時會是追加模式。如果該文件不存在，創建新文件用於讀寫。
rb+	以二進制格式打開一個文件用於讀寫。文件指針將會放在文件的開頭。
wb+	以二進制格式打開一個文件用於讀寫。如果該文件已存在則將其覆蓋。如果該文件不存在，創建新文件。
ab+	以二進制格式打開一個文件用於追加。如果該文件已存在，文件指針將會放在文件的結尾。如果該文件不存在，創建新文件用於讀寫。

<2>關閉文件

close( )

示例如下：

 

   # 新建一個文件，文件名為:test.txt f = open('test.txt', 'w') # 關閉這個文件 f.close()

文件的讀寫

<1>寫數據(write)

使用write()可以完成向文件寫入數據

demo: 新建一個文件 file_write_test.py,向其中寫入如下代碼:

f = open('test.txt', 'w') f.write('hello world, i am here!') f.close() 

運行之后會在file_write_test.py文件所在的路徑中創建一個文件test.txt,其中數據如下:

注意：

• 如果文件不存在那么創建，如果存在那么就先清空，然后寫入數據

追加

　　

f= open('text.txt','a')
f.write('122559阿斯弗')
f.close()

<2>讀數據(read)

使用read(num)可以從文件中讀取數據，num表示要從文件中讀取的數據的長度（單位是字節），如果沒有傳入num，那么就表示讀取文件中所有的數據

demo: 新建一個文件file_read_test.py，向其中寫入如下代碼:

f = open('test.txt', 'r') content = f.read(5) # 最多讀取5個數據 print(content) print("-"*30) # 分割線，用來測試 content = f.read() # 從上次讀取的位置繼續讀取剩下的所有的數據 print(content) f.close() # 關閉文件，這個可以是個好習慣哦 

運行現象：

hello
------------------------------
 world, i am here!

注意：

• 如果用open打開文件時，如果使用的"r"，那么可以省略，即只寫 open('test.txt')

<3>讀數據（readlines）

就像read沒有參數時一樣，readlines可以按照行的方式把整個文件中的內容進行一次性讀取，並且返回的是一個列表，其中每一行的數據為一個元素

#coding=utf-8 f = open('test.txt', 'r') content = f.readlines() print(type(content)) i=1 for temp in content: print("%d:%s" % (i, temp)) i += 1 f.close() 

運行現象：

<4>讀數據（readline）

#coding=utf-8 f = open('test.txt', 'r') content = f.readline() print("1:%s" % content) content = f.readline() print("2:%s" % content) f.close()

二進制數據的讀取rb

 

 

應用1:制作文件的備份

任務描述

• 輸入文件的名字，然后程序自動完成對文件進行備份

參考代碼

# 提示輸入文件 oldFileName = input("請輸入要拷貝的文件名字:") # 以讀的方式打開文件 oldFile = open(oldFileName,'rb') # 提取文件的后綴 fileFlagNum = oldFileName.rfind('.') if fileFlagNum > 0: fileFlag = oldFileName[fileFlagNum:] # 組織新的文件名字 newFileName = oldFileName[:fileFlagNum] + '[復件]' + fileFlag # 創建新文件 newFile = open(newFileName, 'wb') # 把舊文件中的數據，一行一行的進行復制到新文件中 for lineContent in oldFile.readlines(): newFile.write(lineContent) # 關閉文件 oldFile.close() newFile.close()

文件的相關操作

有些時候，需要對文件進行重命名、刪除等一些操作，python的os模塊中都有這么功能

1. 文件重命名

os模塊中的rename()可以完成對文件的重命名操作

rename(需要修改的文件名, 新的文件名)

import os os.rename("畢業論文.txt", "畢業論文-最終版.txt") 

2. 刪除文件

os模塊中的remove()可以完成對文件的刪除操作

remove(待刪除的文件名)

import os os.remove("畢業論文.txt") 

3. 創建文件夾

import os os.mkdir("張三") 

4. 獲取當前目錄

import os os.getcwd() 

5. 改變默認目錄

import os os.chdir("../") 

6. 獲取目錄列表

import os os.listdir("./") 

7. 刪除文件夾

import os os.rmdir("張三")

應用：批量修改文件名

--運行程序之前

 

 運行程序之后

 

 

2>參考代碼

#coding=utf-8 # 批量在文件名前加前綴 import os funFlag = 1 # 1表示添加標志 2表示刪除標志 folderName = './renameDir/' # 獲取指定路徑的所有文件名字 dirList = os.listdir(folderName) # 遍歷輸出所有文件名字 for name in dirList: print name if funFlag == 1: newName = '[東哥出品]-' + name elif funFlag == 2: num = len('[東哥出品]-') newName = name[num:] print newName os.rename(folderName+name, folderName+newName)

綜合應用:學生管理系統(文件版)

import time import os # 定一個列表，用來存儲所有的學生信息(每個學生是一個字典) info_list = [] def print_menu(): print("---------------------------") print(" 學生管理系統 V1.0") print(" 1:添加學生") print(" 2:刪除學生") print(" 3:修改學生") print(" 4:查詢學生") print(" 5:顯示所有學生") print(" 6:保存數據") print(" 7:退出系統") print("---------------------------") def add_new_info(): """添加學生信息""" global info_list new_name = input("請輸入姓名:") new_tel = input("請輸入手機號:") new_qq = input("請輸入QQ:") for temp_info in info_list: if temp_info['name'] == new_name: print("此用戶名已經被占用,請重新輸入") return # 如果一個函數只有return就相當於讓函數結束，沒有返回值 # 定義一個字典，用來存儲用戶的學生信息(這是一個字典) info = {} # 向字典中添加數據 info["name"] = new_name info["tel"] = new_tel info["qq"] = new_qq # 向列表中添加這個字典 info_list.append(info) def del_info(): """刪除學生信息""" global info_list del_num = int(input("請輸入要刪除的序號:")) if 0 <= del_num < len(info_list): del_flag = input("你確定要刪除么?yes or no") if del_flag == "yes": del info_list[del_num] else: print("輸入序號有誤,請重新輸入") def modify_info(): """修改學生信息""" global info_list modify_num = int(input("請輸入要修改的序號:")) if 0 <= modify_num < len(info_list): print("你要修改的信息是:") print("name:%s, tel:%s, QQ:%s" % (info_list[modify_num]['name'], info_list[modify_num]['tel'],info_list[modify_num]['qq'])) info_list[modify_num]['name'] = input("請輸入新的姓名:") info_list[modify_num]['tel'] = input("請輸入新的手機號:") info_list[modify_num]['qq'] = input("請輸入新QQ:") else: print("輸入序號有誤,請重新輸入") def search_info(): """查詢學生信息""" search_name = input("請輸入要查詢的學生姓名:") for temp_info in info_list: if temp_info['name'] == search_name: print("查詢到的信息如下:") print("name:%s, tel:%s, QQ:%s" % (temp_info['name'], temp_info['tel'], temp_info['qq'])) break else: print("沒有您要找的信息....") def print_all_info(): """遍歷學生信息""" print("序號\t姓名\t\t手機號\t\tQQ") i = 0 for temp in info_list: # temp是一個字典 print("%d\t%s\t\t%s\t\t%s" % (i, temp['name'], temp['tel'], temp['qq'])) i += 1 def save_data(): """加載之前存儲的數據""" f = open("info_data.data", "w") f.write(str(info_list)) f.close() def load_data(): """加載之前存儲的數據""" global info_list f = open("info_data.data") content = f.read() info_list = eval(content) f.close() def main(): """用來控制整個流程""" # 加載數據（1次即可） load_data() while True: # 1. 打印功能 print_menu() # 2. 獲取用戶的選擇 num = input("請輸入要進行的操作(數字):") # 3. 根據用戶選擇,做相應的事情 if num == "1": # 添加學生 add_new_info() elif num == "2": # 刪除學生 del_info() elif num == "3": # 修改學生 modify_info() elif num == "4": # 查詢學生 search_info() elif num == "5": # 遍歷所有的信息 print_all_info() elif num == "6": # 保存數據到文件中 save_data() elif num == "7": # 退出系統 exit_flag = input("親,你確定要退出么?~~~~(>_<)~~~~(yes or no) ") if exit_flag == "yes": break else: print("輸入有誤,請重新輸入......") input("\n\n\n按回車鍵繼續....") os.system("clear") # 調用Linux命令clear完成清屏 # 程序的開始 main() 

注意：

• 以上程序，在運行之前請先建立info_data.data文件，並且寫入一對中括號[]即可
• 等到后面講解完異常之后就解決了這個問題