PythonのIO操作

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')