IO操作,即对硬盘上的数据进行读写操作。
直接上代码:
try:
# 'r'表示读取文件,encoding表示编码,errors='ignore'可以忽略错误编码,避免报错UnicodeDecodeError
f = open('F:\\233.txt', 'r', encoding='utf-8', errors='ignore')
# 一次性读取所有内容
# print(f.read())
# 一次最多读取n个字节,这里n=1024
# print(f.read(1024))
# 一次读取一行
# print(f.readline())
# 一次读取所有行,返回list
for line in f.readlines():
# strip()去除末尾的换行符\n
print(line.strip())
finally:
if f:
# 关闭
f.close()
对于读写操作,最后一定要关闭文件对象,因为文件对象会占用操作系统的资源。代码不够简洁,Python支持with语句自动调用close()方法,代码如下:
with open('F:\\233.txt', 'r') as f:
f.read()
如果要读取二进制文件,比如图片、视频等,要用'rb'模式。
with open('F:\\原型图\\Book detail.png', 'rb') as b:
# 输出十六进制表示的字节
b.read()
写文件跟读文件类似,'w'表示写文本文件, 'wb'表示写二进制文件
# 如果abc.txt已经存在,那么会先删除,再重建一个abc.txt
with open('F:\\abc.txt', 'w', encoding='utf-8') as wf:
wf.write('Hello')
如果要在文件末尾追加内容呢?这时候需要'a'模式,代码如下:
with open('F:\\233.txt', 'a', encoding='utf-8') as af:
af.write('Hello, 233')
StringIO类
StringIO类有两种方式往内存中写str。
第一种:
from io import StringIO
# StringIO 在内存中读写str
string_io = StringIO()
string_io.write('Hello')
string_io.write(', ')
string_io.write('World')
# 获取写入的str
print('StringIO: ', string_io.getvalue())
第二种:
s = StringIO('Hello!\n Read! ')
while True:
# 可以像读文件一样读取内容
line = s.readline()
if not line:
break
print(line.strip())
BytesIO类
跟StringIO类似,只不过StringIO是写入str,BytesIO是写入byte。
这里需要注意getvalue()方法和read()方法因为stream position而产生的差异,可以通过tell()获取position,也可以通过seek()调整position。
from io import BytesIO
# 在内存中读写二进制数据
bytes_io = BytesIO()
bytes_io.write('天朝'.encode('utf-8'))
print('position已经移到%d' % bytes_io.tell())
print(bytes_io.getvalue().decode('utf-8'))
# write后,stream position已经移到6,所以接下来read读取不到内容
print('seek之前,', bytes_io.read().decode('utf-8'))
# seek调整stream position
bytes_io.seek(0)
print('seek之后,', bytes_io.read().decode('utf-8'))
b = BytesIO('天朝'.encode('utf-8'))
# 此时stream position为0
print('position已经移到%d' % b.tell())
print(b.read().decode('utf-8'))
OS类
import os
# nt代表windows,posix代表Linux、UNIX或Mac OS
print(os.name)
当前目录的绝对路径
print(os.path.abspath('.'))
当前目录的相对路径
print(os.path.relpath('F:/pythonCode/pythonCode.iml', 'F:'))
如果要在某个目录下创建一个新目录,首先获取新目录的完整路径(此时并没有创建出目录)
path_join = os.path.join('F:/Z', 'ZZZZ')
判断目录是否存在
if os.path.exists(path_join):
# 删除目录
os.rmdir(path_join)
创建目录
os.mkdir(path_join)
os.path.split把路径拆分为两部分,后一部分一定是最后级别的目录或者文件名
print(os.path.split(path_join))
print(os.path.split('F:/233.txt'))
输出:
('F:/Z', 'ZZZZ')
('F:/', '233.txt')
os.path.splitext把路径拆分为两部分,后一部分一定是文件后缀名或者''
print(os.path.splitext(path_join))
print(os.path.splitext('F:/666.txt'))
输出:
('F:/Z\\ZZZZ', '')
('F:/666', '.txt')
上面合并、拆分的函数并不要求目录和文件一定存在,它们只对字符串操作。
重命名文件
os.rename('F:/555.txt', 'F:/888.txt')
删除文件
os.remove('F:/888.txt')
复制移动文件等功能在shutil模块中
import shutil
# 复制文件
shutil.copy('F:/233.txt', 'F:/Z/233.txt')
# 移动文件
# shutil.move('F:/666.txt', 'F:/Z/666.txt')
获取当前目录下所有目录名,不包含目录的路径
print([x for x in os.listdir('.') if os.path.isdir(x)])
获取当前目录下所有后缀名为.py的文件名,不包含文件的路径
print([x for x in os.listdir('.') if os.path.isfile(x) and os.path.splitext(x)[1] == '.py'])
获取某个目录下文件名包含'Test'字符串的文件相对路径:
def get_file(path='.'):
if os.path.exists(path):
if os.path.isdir(path):
dirs = os.listdir(path)
for v in dirs:
# v只是当前目录下的目录名或者文件名,需重新组合路径
join = os.path.join(path, v)
if os.path.isfile(v):
if v.find('Test') != -1:
print(join)
else:
get_file(join)
else:
if path.find('Test') != -1:
print(path)
else:
print('path is not exists.')
# 调用函数,传入目录
get_file('F:\\pythonCode')