Python读取写入获取文件简单例子
# 对个获取指定目录的所有文件
def get_all_files(dir):
fileList = []
"""遍历获取指定文件夹下面所有文件"""
if os.path.isdir(dir):
filelist = os.listdir(dir)
for ret in filelist:
filename = dir + "\\" + ret
if os.path.isfile(filename):
fileList.append(filename)
return fileList
# 对个获取指定目录的所有文件
def get_file_list_by_walk(dir):
fileList = []
"""使用listdir循环遍历"""
if not os.path.isdir(dir):
return fileList
dirlist = os.walk(dir)
for root, dirs, files in dirlist:
for fi in files:
fileList.append(os.path.join(root, fi))
return fileList
#指定文件路径获取文件最后文件的路径包含文件
#如:D:\test\file.txt 返回的结果为:D:\test\
def get_file_path(path):
#获取文件名
#return os.path.split(path)[1]
# 获取文件路径
return os.path.split(path)[0]
#指定文件路径获取文件最后文件的路径包含文件
#如:D:\test\file.txt 返回的结果为:file.txt
def get_file_name(path):
return os.path.basename(path)
# 创建目录
def mkdir(path):
# 去除尾部 \ 符号
pathx = path.strip().rstrip("\\")
# 判断路径是否存在
# 存在 True
# 不存在 False
isExists = os.path.exists(pathx)
# 判断结果
if not isExists:
# 如果不存在则创建目录创建目录操作函数
os.makedirs(path)
print(path + ' 创建成功')
return True
else:
# 如果目录存在则不创建,并提示目录已存在
print(path + ' 目录已存在')
return False
# 创建一个txt文件,并向文件写入msg
"""
@file_dir参数 代表文件目录 如:D:\test
@file_name参数 代表文件名称 如:file.txt
@msg参数 代表要写入文件的内容信息
"""
def writer_to_file(file_dir, file_name, msg):
# 先创建目录
mkdir(file_dir)
# 再打开文件
full_path = file_dir + "\\" + file_name
fi = open(full_path, 'w')
# 写入文件
fi.write(msg)
fi.close()
"""
通过指定文件路径文件进行读取内容
如:D:\test\file.txt
"""
def reader_file(path):
#解决乱码问题
fi = open(path,'r',encoding='gbk',errors = 'ignore')
strs=[]
#splitlines解决不换行\n输出
for line in fi.read().splitlines():
if(len(line)>0):
strs.append(line)
return strs
#测试__main__方法
if __name__ == '__main__':
path = r"D:\test\test.txt"
#读取内容
fileStr = reader_file(path)
# 最终统计的信息写入文件
writer_to_file(path, "file.txt", fileStr)
print('do finish')