指定一种后缀:
import os
import os.path
ls = []
def getAppointFile(path, ls, hz):
fileList = os.listdir(path)
try:
for tmp in fileList:
pathTmp = os.path.join(path, tmp)
if True==os.path.isdir(pathTmp):
getAppointFile(pathTmp, ls, hz)
elif pathTmp[pathTmp.rfind('.') + 1:].upper() == hz.upper():
ls.append(pathTmp)
except PermissionError:
pass
def main():
while True:
path = input('输入指定路径: ').strip()
hz = input('输入相同后缀: ').strip()
if os.path.isdir(path) == True:
break
getAppointFile(path, ls, hz)
for lss in ls:
print(lss)
print(len(ls))
main()
多个后缀文件:
import os
def search_file(start_dir, target) :
os.chdir(start_dir)
for each_file in os.listdir(os.curdir) :
ext = os.path.splitext(each_file)[1]
if ext in target :
vedio_list.append(os.getcwd() + os.sep + each_file + os.linesep)
if os.path.isdir(each_file) :
search_file(each_file, target) # 递归调用
os.chdir(os.pardir) # 递归调用后切记返回上一层目录
start_dir = input('请输入待查找的初始目录:')
program_dir = os.getcwd()
target = ['.mp4', '.avi', '.rmvb']
vedio_list = []
search_file(start_dir, target)
f = open(program_dir + os.sep + 'vedioList.txt', 'w')
f.writelines(vedio_list)
f.close()
获取文件后缀:
def getfile_fix(filename):
return filename[filename.rfind('.') + 1:]
print(getfile_fix('runoob.txt'))
指定文件中的内容替换:
def file_replace(file_name, rep_word, new_word):
f_read = open(file_name)
content = []
count = 0
for eachline in f_read:
if rep_word in eachline:
count = count+eachline.count(rep_word)
eachline = eachline.replace(rep_word, new_word)
content.append(eachline)
decide = input('\n文件 %s 中共有%s个【%s】\n您确定要把所有的【%s】替换为【%s】吗?\n【YES/NO】:' \
% (file_name, count, rep_word, rep_word, new_word))
if decide in ['YES', 'Yes', 'yes']:
f_write = open(file_name, 'w')
f_write.writelines(content)
f_write.close()
f_read.close()
file_name = input('请输入文件名:')
rep_word = input('请输入需要替换的单词或字符:')
new_word = input('请输入新的单词或字符:')
file_replace(file_name, rep_word, new_word)
原文地址:http://www.runoob.com/python3/python3-file-methods.html