yoyo2 发表于 2020-8-23 18:44:28

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

本帖最后由 yoyo2 于 2020-8-23 18:48 编辑

'''
编写一个程序,这次要求使用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 发表于 2020-8-23 19:24:45



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

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

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

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

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

yoyo2 发表于 2020-8-23 22:58:33

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

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

噢,原来是这样,多谢,我split理解错了
页: [1]
查看完整版本: python基础第31讲pickle模块动动手