python基础第31讲pickle模块动动手

yoyo2

'''
编写一个程序，这次要求使用pickle将record.txt文件里的对话按照以下要求
腌制成不同文件（没错，是第29讲的内容小改，考考你自己能写出来吗？）：

    小甲鱼的对话单独保存为boy_*.txt的文件（去掉“小甲鱼:”）
    小客服的对话单独保存为girl_*.txt的文件（去掉“小客服:”）
    文件中总共有三段对话，分别保存为boy_1.txt, girl_1.txt，boy_2.txt, girl_2.txt, boy_3.txt, gril_3.txt共6个文件（提示：文件中不同的对话间已经使用“==========”分割）
'''
import pickle
import os

os.chdir('D:\\workspace')
 #一次读一行
#如果内容不是=========
#-----以“ ：”分离，如果前面是小甲鱼就放boy[]，如果前面是小客服就放[]
#如果内容是=========
#-----保存数据;count += 1;boy[]和girl[]清零
#因为只有两行========，所以要还有执行保存数据一次
def save(boy,girl,count):
    boy_file = 'boy_' + str(count) + '.txt'#文件名
    girl_file = 'girl_' + str(count) + '.txt'#文件名

    file_save = open(boy_file,'wb')
    pickle.dump(boy,file_save)
    file_save.close()

    file_save = open(girl_file,'wb')
    pickle.dump(girl,file_save)
    file_save.close()

def split_file(file):
    count = 1
    boy = []
    girl = [] 
    
    file = open(file) 

    for each_line in file:
        if each_line[:6] != '======': #报错处1,原代码为if each_line != '========='会无法判断========行
            (person, talk) = each_line.split(':',1)#报错处2，原代码为(person, talk) = each_line.split(':')原因是有一行有两个“ ：”系统无法判断怎么分割
            if person == '小甲鱼':
                boy.append(talk)
            elif person == '小客服':
                girl.append(talk)
        else:
            save(boy,girl,count)

            boy.clear()
            girl.clear()

            count += 1
    save(boy,girl,count)#漏这步只能产生两个文件，因为只有两行“=========”

split_file('D:\\workspace\\record.txt')

Twilight6

这里的报错处 1 和 2 实际上应该只有一个

这里  if each_line != '=========' 应该不会报错，而报错原因都应该是第二个

这里是因为当你直接 if each_line != '=========' 时候，Python 按行读取txt 文本时候会自带 \n 换行符，所以 if 条件永远都会成立

导致 split 切割这行都是等号的时候，因为没有冒号，所以 split 只返回含有一个元素的列表，而你设置了两个变量接受切割后的列表元素，所以导致的报错

当你代码改成 (person, talk) = each_line.split(':') 同理，并不是因为系统无法判断怎么分隔，是因为系统无法分配切割后的元素

yoyo2

Twilight6 发表于 2020-8-23 19:24
这里的报错处 1 和 2 实际上应该只有一个

这里  if each_line != '=========' 应该不会报错，而报错 ...

噢，原来是这样，多谢，我split理解错了