2020-02-09 Python文本处理

1. Python编码解码

将文本转换为二进制数据就是编码,将二进制数据转换为文本就是解码。编码和解码要按照一定的规则进行,这个规则就是字符集。

# -*- coding: utf-8 -*-
# 本文件应该保存为utf-8编码,否则会报错

str = "中文测试"
print(f'Unicode字符串为"{str}"')
byte0 = str.encode("utf-8")
print(f'Unicode字符串"{str}"以utf-8编码得到字节串[{byte0}]')
str0 = byte0.decode("utf-8")
print(f'将utf-8字节串[{byte0}]解码得到字符串"{str0}"')
byte1 = str.encode("gbk")
print(f'Unicode字符串"{str}"以gbk编码得到字节串[{byte1}]')
str1 = byte1.decode("gbk")
print(f'将gbk字节串[{byte1}]解码得到Unicode字符串"{str1}"')

print(f'以文本方式将Unicode字符串"{str}"写入a.txt')

with open("a.txt", "w", encoding="gbk") as f:
    f.write(str)

print("以文本方式读取a.txt的内容")
with open("a.txt", "r", encoding="gbk") as f:
    print(f.read())
代码示例运行结果

默认编码可以通过sys.getdefaultencoding()来查看Python解释器会用的默认编码:

import sys
encoding = sys.getdefaultencoding()
print(f"{encoding}")

2. 文件操作

2.1 普通文件操作

f = open("a.txt")   #打开文件,得到一个文件句柄,并赋值给一个变量
print(f.read())     #打印读取文件的内容
f.close()           #关闭文件

open函数原型如下:

open(file, mode='r', buffering=-1, encoding=None, errors=None, newline=None, closefd=True, opener=None)
其中:
(1)参数file是一个表示文件名称的字符串,如果文件不在程序当前的路径下,就需要在前面加上相对路径或绝对路径。
(2)参数mode是一个可选字参数,指示打开文件的方式,若不指定,则默认为以读文本的方式打开文件。
字符串及含义

默认的打开方式是'rt'(mode='rt')。Python是区分二进制方式和文本方式的,当以二进制方式打开一个文件时(mode参数后面跟'b'),返回一个未经解码的字节对象:当以文本方式打开文件时(默认是以文本方式打开,也可以mode参数后面跟't'),返回一个按系统默认编码或参数encoding传入的编码来解码的字符串对象。
(3)buffering是一个可选的参数,buffering=0表示关闭缓冲区(仅在二进制方式打开时可用):buffering=1表示选择行缓冲区(仅在文本方式打开时可用):buffering大于1时,其值代表固定大小的块缓冲区的大小。当不指定该参数时,默认的缓冲策略是这样的:二进制文件使用固定大小的块缓冲区,文本文件使用行缓冲区。

# -*- coding: utf-8 -*-

f = open("wb.txt", "w", encoding="utf-8")
f.write("测试w方式写入,如果文件存在,则清空内容后写入:如果文件不存在,则创建\n")
f.close()

f = open("wb.txt", "a", encoding="utf-8")
f.write("测试a方式写入,如果文件存在,则在文件内容后最后追加写入;如果文件不存在,则创建")
f.close()

f = open("wb.txt", "r", encoding="utf-8")
# 以文本方式读,f.read()返回字符串对象
data = f.read()
print(type(data))
print(data)
f.close()

f = open("wb.txt", "rb")
# 以文本方式读,f.read()
data = f.read()
print(type(data))
print(data)
print('将读取的字符对象解码:')
print(data.decode('utf-8'))
f.close()
运行结果

从上面的例子可以看出,以二进制读取文件时,读取的是文件字符串的编码(以encoding指定的编码格式进行的编码),将读取的字节对象解码,可以得出原字符串。
需要注意以下几点:
(1)记得使用完毕后及时关闭文件,释放资源,如f.close(),否则就会导致操作系统打开的文件还没有关闭,白白占用资源。使用with关键字来帮我们管理上下文,系统会自动为我们关闭文件和处理异常,如下面两行代码即可完成安全的写操作。

with open('a.txt', 'w') as f:
    f.write("hello word")

(2)open()函数是由操作系统打开文件,如果我们没有为open指定编码,那么打开文件的默认编码很明显是操作系统默认的编码:在Windows下是gbk,在Linux下是utf-8。若要保证不乱码,就必须让读取文件和写入文件使用的编码一致。
常见的文件操作方法
f.readable():判断有读取文件的方法
f.read([size]):size为读取的长度,以byte为单位,如果未给定或为负则读取所有
f.readline([size]) :读取整行,包括 "\n" 字符
f.readlines([size]) : 返回一个list,其实内部是调用readline()实现的
f.write(str) : 把str写到文件中,write()不主动加换行符
f.writelines(str) : 把多行内容一次写入
f.close() : 关闭文件
f.flush() : 把缓冲区的内容写入硬盘
f.tell() : 当前文件操作标记的位置
f.next() : 返回下一行,并将文件操作标记位移到下一行。把一个file用于for … in file这样的语句时,就是调用next()函数来实现遍历的
f.seek(offset[,whence]) :将文件打操作标记移到offset的位置。这个offset一般是相对于文件的开头来计算的,一般为正数。whence参数,whence可以为0表示从头开始计算,1表示以当前位置为原点计算。2表示以文件末尾为原点进行计算。需要注意,如果文件以a或a+的模式打开,每次进行写操作时,文件操作标记会自动返回到文件末尾

在文件中定位示例:
f = open("tmp.txt", "rb+")
a = f.write(b"abcdefghi")
print(f"一共有{a}个字节")

b = f.seek(5)   #移动到文件的第六个字节
print(f"第六个字节:{b}")
print(f.read(1))

c = f.seek(-3,2)    #移动到文件的倒数第三个字节
print(f"倒数第三个字节:{c}")
print(f.read(1))
定位结果

基于seek实现类似Linux命令tail -f的功能:

# encoding=utf-8                                    
                                                    
import time                                         
                                                    
with open('tmp.txt', 'rb') as f:                    
    f.seek(0, 2)            # 将光标移动到文件末尾            
    while True:             # 实时显示文件新增加的内容          
        line = f.read()                             
        if line:                                    
            print(line.decode('utf-8'), end='')     
        else:                                       
            time.sleep(0.2)         # 读取完毕后短暂的睡眠    

当tmp.txt追加新的内容时,新内容会被程序立刻打印出来。

2.2 大文件的读取

当文件较小时,我们可以一次性全部读入内存,对文件的内容作出任意修改,再保存至磁盘,这一过程会非常快。
将文件a.txt中的字符串str1替换为str2

with open('a.txt') as read_f,open('.a.txt.swap', 'w') as write_f:
    data=read_f.read()      #全部读入内存,如果文件很大,则会很卡
    data=data.replace('str1', 'str2')       #在内存中完成修改
    
    write_f.write(data)     #一次性写入新文件
    
os.remove('a.txt')
os.rename('.a.txt.swap', 'a.txt')

当文件很大时,如GB级的文本文件,上面的代码运行将会非常缓慢,此时我们需要使用文件的可迭代方式将文件的内容逐行读入内存,再逐行写入新文件,最后用新文件覆盖源文件。
对大文件进行读写

import os
with open('a.txt') as read_f,open('.a.txt.swap','w') as write_f:
    for line in read_f:     #对可迭代对象f逐行操作,防止内存溢出
        line=line.replace('str1', 'str2')
        write_f.write(line)
os.remove('a.txt')
os.rename('.a.txt.swap', 'a.txt')

大文件为a.txt,当打开文件时,会得到一个可迭代对象read_f,对可迭代对象进行逐行读取,可防止内存溢出,也会加快处理速度。
处理大数据还有多种方法:
(1)通过read(size)增加参数,指定读取的字节数。

while True:
    block = f.read(1024)
    if not block:
        break

(2)通过readline,每次只读一行。

while True:
    line = f.readline()
    if not line:
        break
file对象常用的函数

3. 序列化和反序列化

  • 序列化:将数据结构或对象转换成二进制串的过程。
  • 反序列化:将在序列化过程中所生成的二进制串转换成数据结构或对象的过程。
    序列化实例
# encoding:utf-8

import pickle

# 使用pickle模块将数据对象保存到文件

# 字符串
data0 = "hello world"
# 列表
data1 = list(range(20))[1::2]
# 元组
data2 = ("x", "y", "z")
# 字典
data3 = {"a": data0, "b": data1, "c": data2}

print(data0)
print(data1)
print(data2)
print(data3)

output = open("data.pkl", "wb")

# 使用默认的protocol
pickle.dump(data0, output)
pickle.dump(data1, output)
pickle.dump(data2, output)
pickle.dump(data3, output)
output.close()
运行结果

反序列化实例

# encoding=utf-8

import pickle

# 使用pickle模块从文件中重构Python对象
pkl_file = open("data.pkl", "rb")

data0 = pickle.load(pkl_file)
data1 = pickle.load(pkl_file)
data2 = pickle.load(pkl_file)
data3 = pickle.load(pkl_file)

print(data0)
print(data1)
print(data2)
print(data3)

pkl_file.close()
运行结果

可以看出运行结果与序列化实例运行的结果完全一致。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 202,980评论 5 476
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,178评论 2 380
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 149,868评论 0 336
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,498评论 1 273
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,492评论 5 364
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,521评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,910评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,569评论 0 256
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,793评论 1 296
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,559评论 2 319
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,639评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,342评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,931评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,904评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,144评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,833评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,350评论 2 342

推荐阅读更多精彩内容