(給Python開發者加星標,提升Python技能)
來源:詭途
https://blog.csdn.net/qq_35866846/article/details/107823636
【導語】:有沒有想過用python寫一個文件管理程序?聽起來似乎沒思路?其實是可以的,因為python已經為你準備好了神器os.walk,進來看看吧!
python中os.walk是一個簡單易用的文件、目錄遍歷器,可以幫助我們高效的處理文件、目錄方面的事情。
本文將詳細介紹os.walk模塊,最後使用os.walk模塊實現一個在指定日誌整理文件的程序。
基本介紹
os.walk():掃描某個指定目錄下所包含的子目錄和文件,返回的是一個迭代器。
基本使用
假設文件夾data有如下的目錄結構(cmd 命令:tree /f)
2.1掃描所有文件
掃描內容:
輸出內容:
掃描路徑:
自頂向下 topdown=True(默認)
自底向上 topdown=False
from os import walkpath="data"for curDir, dirs, files in walk(path):#for curDir, dirs, files in walk(path,topdown=False): print("現在的目錄:" ,curDir) print("該目錄下包含的子目錄:" , str(dirs)) print("該目錄下包含的文件:",str(files)) print("*"*20)自頂向下掃描結果:
現在的目錄:data該目錄下包含的子目錄:['testA', 'testB', 'testC']該目錄下包含的文件:['2020-07-12 - 第一層.xlsx', '2020-07-13 - 第一層.xlsx', '2020-07-14 - 第一層.xlsx']********************現在的目錄:data\testA該目錄下包含的子目錄:[]該目錄下包含的文件:['2020-07-12-A.xlsx', '2020-07-13-A.xlsx', '2020-07-14-A.xlsx']********************現在的目錄:data\testB該目錄下包含的子目錄:[]該目錄下包含的文件:['2020-07-12-B.xlsx', '2020-07-13-B.xlsx', '2020-07-14-B.xlsx']********************現在的目錄:data\testC該目錄下包含的子目錄:[]該目錄下包含的文件:['2020-07-12-C.xlsx', '2020-07-13-C.xlsx', '2020-07-14-C.xlsx']********************自底向上掃描結果:
現在的目錄:data\testA該目錄下包含的子目錄:[]該目錄下包含的文件:['2020-07-12-A.xlsx', '2020-07-13-A.xlsx', '2020-07-14-A.xlsx']********************現在的目錄:data\testB該目錄下包含的子目錄:[]該目錄下包含的文件:['2020-07-12-B.xlsx', '2020-07-13-B.xlsx', '2020-07-14-B.xlsx']********************現在的目錄:data\testC該目錄下包含的子目錄:[]該目錄下包含的文件:['2020-07-12-C.xlsx', '2020-07-13-C.xlsx', '2020-07-14-C.xlsx']********************現在的目錄:data該目錄下包含的子目錄:['testA', 'testB', 'testC']該目錄下包含的文件:['2020-07-12 - 第一層.xlsx', '2020-07-13 - 第一層.xlsx', '2020-07-14 - 第一層.xlsx']********************2.2掃描輸出所有文件的路徑
輸出所有文件:import ospath="data"for curDir, dirs, files in os.walk(path): for file in files: print(os.path.join(curDir, file))data\2020-07-12 - 第一層.xlsxdata\2020-07-13 - 第一層.xlsxdata\2020-07-14 - 第一層.xlsxdata\testA\2020-07-12-A.xlsxdata\testA\2020-07-13-A.xlsxdata\testA\2020-07-14-A.xlsxdata\testB\2020-07-12-B.xlsxdata\testB\2020-07-13-B.xlsxdata\testB\2020-07-14-B.xlsxdata\testC\2020-07-12-C.xlsxdata\testC\2020-07-13-C.xlsxdata\testC\2020-07-14-C.xlsx輸出指定類型文件
#endswith 截取文件後綴import ospath="data"for curDir, dirs, files in os.walk(path): [print(os.path.join(curDir, file)) for file in files if file.endswith(".xlsx")]2.3掃描輸出所有的子目錄(子文件夾)
# 使用os.walk輸出某個目錄下的所有文件import ospath="data"for curDir, dirs, files in os.walk(path): for _dir in dirs: print(os.path.join(curDir, _dir))data\testAdata\testBdata\testC案例代碼
#綜合運用os.walk()——文件指定日期整理程序
import pandas as pdimport numpy as npimport os,openpyxldef move_file(file_path,_new_path,date_xl_str): for curDir, dirs, files in os.walk(file_path): for file in files: old_path = os.path.join(curDir, file) new_path = os.path.join(_new_path, file) file_date=file.split("_")[-1][:10] try: os.rename(old_path,new_path) if file_date in date_xl_str else os.remove(old_path) except: os.remove(old_path)
for curDir, dirs, files in os.walk(file_path): for _dir in dirs: os.removedirs(os.path.join(curDir, _dir)) os.mkdir("data")
def qch_date(file_path): wj_names=os.listdir(file_path) wj_list=[] num=0 for wj in wj_names: new_wj=wj[:-11] if new_wj not in wj_list: wj_list.append(new_wj) else: os.remove(file_path+"\\"+wj) num+=1 return num
def refresh_data(file_path,sheet_name,data): book=openpyxl.load_workbook(file_path) writer=pd.ExcelWriter(file_path,engine="openpyxl") writer.book=book ws=book[sheet_name] for row in ws.iter_rows(): for cell in row: cell.value=None idx_num,col_num=data.shape for i in range(1,idx_num+1): for j in range(1,col_num+1): ws.cell(row=i,column=j).value=data.iloc[i-1,j-1] writer.save() writer.close() return None
def check_file(file_path,check_file="文件檢查.xlsx"): wj_names=os.listdir(file_path) data=pd.DataFrame([wj.split("_")[2:] for wj in wj_names],columns=["店鋪名稱","日期"]) data['日期']=data['日期'].str[:10] nind=data.index.insert(0,'0') data1=data.reindex(index=nind) data1.loc['0']=data.columns data1.reset_index(drop=True,inplace=True)
refresh_data(check_file,"數據源",data1) return None
file_path="data"start_date=input("請輸入開始日期:")end_date=input("請輸入開始日期:")
date_xl_str=[str(i)[:10] for i in pd.date_range(start_date,end_date,freq='D')]
new_path=start_date+"~"+end_datetry: os.mkdir(new_path)except: print("文件夾 【%s】 已存在"%new_path) move_file(file_path,new_path,date_xl_str) num=qch_date(new_path)print("去除重複文件 %s 個"%num)
check_file(new_path)- EOF -
覺得本文對你有幫助?請分享給更多人
關注「Python開發者」加星標,提升Python技能
點讚和在看就是最大的支持❤️