Python 文件数据读写的具体实现

脚本专栏 2024/11/25 佚名

3 2 1

文件数据读写

读写文件，本质上是请求操作系统打开一个文件对象，然后，通过操作系统提供的接口从这个文件对象中读取数据（读文件），或者把数据写入这个文件对象（写文件）。

文件读取

使用 Python 内置 open() 函数，以 rt 的模式读取文件，如下示例：

> f = open('some.txt', 'rt')

这行代码就表示打开一个文件，若是文件不存在，会抛出 IOError 的异常，并给出详细的信息提示：

> f = open('undefined.txt', 'rt')
Traceback (most recent call last):
 File "<stdin>", line 1, in <module>
FileNotFoundError: [Errno 2] No such file or directory: 'undefined.txt'

当成功打开文件时，可使用 read() 函数读取文件的内容:

> f.read()
'Hello world!'

当数据读取完毕后，需要调用 close() 关闭文件。因为文件对象会占用资源，使用完毕后需要及时关闭释放资源。

> f.close()

还有一种方法就是使用 with 语句，给被使用的文件创建一个上下文环境，这样文件对象就能够自动关闭。

> with open('some.txt', 'rt') as f:
...   data = f.read()
...

调用 read() 时一次性读取全部内容，若是文件内容过大，可以使用 read(size) 固定大小循环读取，每次最多读取 size 字节的内容。readline() 函数，每次读取一行内容，readlines() 读取所有内容，但是按行返回 list。三者的使用，可根据实际的需求进行选择。

二进制文件

读取二进制的文件，需要使用 rb 的模式打开：

> f = open('image.jpg', 'rb')
> f.read()
b'\xff\xd8\xff\xe0\x00\x10JFIF\x00...'

从文件中读取数据，需要注意编码的问题。当编码错误时，会抛出 UnicodeDecodeError 异常。比如：

> f = open('some.txt', 'rt', encoding='ascii')
> f.read()
Traceback (most recent call last):
 File "<stdin>", line 1, in <module>
 File "/usr/local/lib/python3.6/encodings/ascii.py", line 26, in decode    
  return codecs.ascii_decode(input, self.errors)[0]
UnicodeDecodeError: 'ascii' codec can't decode byte 0xcc in position 2: ordinal not in range(128)

这种情况下，通常为读取文本指定的编码不正确，需要确认文件编码是否正确。如果编码错误还是存在的话，可以给 open() 函数传递一个可选的 errors 参数来处理这些错误。比如：

> f = open('some.txt', 'rt', encoding='ascii', errors='replace') 
> f.read()
'Un"color: #ff0000">压缩文件


读取 gzip 和 bz2 格式的压缩文件时，可以使用 gzip 和 bz2 模块。两个模块都为 open() 提供了另外的实现来解决读取 gzip 和 bz2 两个格式的压缩文件的问题。例如读取压缩文件，示例如下：


> import gzip
> with gzip.open('some.gz', 'rt') as f: 
...   text=f.read()
   
> import bz2
> with bz2.open('some.bz2', 'rt') as f:
...   text=f.read()



文件写入


文件写入，同样需要调用 open() 参数，但是指定的模式为 wt 或者 wb，用以表示写文本文件或是二进制文件：


> f = open('some.txt', 'wt')
> f.write('Hello, world!')
13
> f.close()



数据写入完毕，同样需要调用 close() 关闭文件对象。也可以 with 语句创建上下文，用以正常关闭文件对象。
类似的，写入压缩数据时，导入 gzip 或者 bz2 模块：


> import gzip
> with gzip.open('some.gz', 'wt') as f: 
...   f.write(text)
   
> import bz2
> with bz2.open('some.bz2', 'wt') as f:
...   f.write(text)



写入压缩数据时，可选参数 compresslevel 可以指定一个压缩等级。例如：


> with gzip.open('some.gz', 'wt', compresslevel=6) as f:
...   f.write(text)



compresslevel 参数默认值为 9，表示最高的压缩等级。等级越低性能越好，但是压缩程度也就越低。
使用 w 模式的 open() 函数，若打开的文件在写入前有数据内容，则会被清除覆盖。如果是要在已经存在的文件中添加内容的话，使用模式为 at 的 open() 函数。
涉及读写文件模式的定义及含义，内容如下：


  
    
      模式
      含义
    
  
  
    
      'r'
      读取数据（默认）
    
    
      'w'
      写入数据
    
    
      'x'
      文件存在抛出异常，不存在则创建
    
    
      'a'
      追加写入数据
    
    
      'b'
      二进制模式
    
    
      't'
      文本模式（默认）
    
    
      '+'
      更新文件（读写）
    
  


具体更详细的内容可以参考 Python 官方文档。
参考资料


来源

David M. Beazley;Brian K. Jones.Python Cookbook, 3rd Edtioni.O'Reilly Media.2013.

Luciano Ramalho.Fluent Python.O'Reilly Media.2015.

“2. Built-in Functions”.docs.python.org.Retrieved 20 January 2020.

"gzip — Support for gzip files".docs.python.org.Retrieved 22 January 2020.

"bz2 — Support for bzip2 compression".docs.python.org.Retrieved 22 January 2020.

廖雪峰.“Python 教程”.liaoxuefeng.com.[2020-01-18].


以上为本篇的主要内容。希望对大家的学习有所帮助，也希望大家多多支持。