029文件：一个任务

人若有情死得早

任务：将文件（record.txt）中的数据进行分割并按照以下规律保存起来：
-小甲鱼的对话单独保存为boy_*.txt的文件（去掉“小甲鱼：”）
-小客服的对话单独保存为girl_*.txt的文件（去掉“小客服：”）
-文件中总共有三段对话，分别保存为boy_1.txt, girl_1.txt, boy_2.txt, girl_2.txt, boy_3.txt, girl_3.txt共6个文件（提示：文件中不同的对话间已经使用“======”分割）

1. f = open('record.txt')
   
2. 
   
3. 
   
4. boy = []
   
5. girl = []
   
6. count = 1
   
7. 
   
8. for each_line in f:
   
9.     if each_line[:6] != '======':
   
10.        (role, line_spoken) = each_line.split(':',1)
    
11.        if role == '小甲鱼':
    
12.            boy.append(line_spoken)
    
13.        if role == '小客服':
    
14.             girl.append(line_spoken)
    
15.     else:
    
16.         file_name_boy = 'boy_' + str(count) + '.txt'
    
17.         file_name_girl = 'girl_' + str(count) + '.txt'
    
18. 
    
19.         boy_file = open(file_name_boy, 'w')
    
20.         girl_file = open(file_name_girl,'w')
    
21. 
    
22.         boy_file.writelines(boy)
    
23.         girl_file.writelines(girl)
    
24. 
    
25.         boy_file.close()
    
26.         girl_file.close()
    
27. 
    
28.         boy = []
    
29.         girl = []
    
30.         count += 1
    
31. 
    
32. file_name_boy = 'boy_' + str(count) + '.txt'
    
33. file_name_girl = 'girl_' + str(count) + '.txt'
    
34. 
    
35. boy_file = open(file_name_boy, 'w')
    
36. girl_file = open(file_name_girl,'w')
    
37. 
    
38. boy_file.writelines(boy)
    
39. girl_file.writelines(girl)
    
40. 
    
41. boy_file.close()
    
42. girl_file.close()
    
43. 
    
44. f.close()

复制代码

代码优化

1. def save_file(boy, girl, count):
   
2.     file_name_boy = 'boy_' + str(count) + '.txt'
   
3.     file_name_girl = 'girl_' + str(count) + '.txt'
   
4. 
   
5.     boy_file = open(file_name_boy, 'w')
   
6.     girl_file = open(file_name_girl,'w')
   
7. 
   
8.     boy_file.writelines(boy)
   
9.     girl_file.writelines(girl)
   
10. 
    
11.     boy_file.close()
    
12.     girl_file.close()
    
13. 
    
14. def split_file(file_name):
    
15.     f = open('record.txt')
    
16. 
    
17. 
    
18.     boy = []
    
19.     girl = []
    
20.     count = 1
    
21. 
    
22.     for each_line in f:
    
23.         if each_line[:6] != '======':
    
24.            (role, line_spoken) = each_line.split(':',1)
    
25.            if role == '小甲鱼':
    
26.                boy.append(line_spoken)
    
27.            if role == '小客服':
    
28.                 girl.append(line_spoken)
    
29.         else:
    
30.             save_file(boy, girl, count)
    
31. 
    
32.             boy = []
    
33.             girl = []
    
34.             count += 1
    
35. 
    
36.     save_file(boy, girl, count)
    
37. 
    
38.     f.close()
    
39. 
    
40. 
    
41. split_file('record.txt')

复制代码