《零基础入门学习Python》第二十九讲学习笔记

一只小章鱼

1、本讲主要是说文件操作

f=open('recore.txt')
boy=[]
girl=[]
count = 1
for each in f:
      if each[:6] != '=====':
           (role,line_spoken)=each.split(':',1)
           if role == '小甲鱼':
                boy.append(line_spoken)
           if role == '小客服':
                girl.append(line_spoken)
      else:
                file_name_boy = 'boy_' + str(count) +'.txt'
                file_name_girl = 'girl_' + str(count) +'.txt'
 
                boy_file = open(file_name_boy,'w')
                girl_file =  open(file_name_girl,'w')
 
                boy_file.writelines(boy)
                girl_file.writelines(girl)
  
                boy_file.close()
                girl_file.close()
          
                boy=[]
                girl = []
                count += 1

file_name_boy = 'boy_' + str(count) +'.txt'
file_name_girl = 'girl_' + str(count) +'.txt'
 
boy_file = open(file_name_boy,'w')
girl_file =  open(file_name_girl,'w')
 
boy_file.writelines(boy)
girl_file.writelines(girl)

f.close()

可以将重复的代码进行封装打包变成一个函数