字典与集合应用

Cool_Breeze

将此文本按照相同用户每次只输出3行数据（第一列为用户），不满足用户最后全部输出。

1. 1005495524----a111111
   
2. 1005495524----abc123
   
3. 1005495524----aini1314
   
4. 1005495524----iloveyou
   
5. 1005495524----q1w2e3r4
   
6. 1005495524----qq123123
   
7. 1044122462aa----a111111
   
8. 1044122462aa----abc123
   
9. 1044122462aa----aini1314
   
10. 1044122462aa----iloveyou
    
11. 1044122462aa----q1w2e3r4
    
12. 1044122462aa----qq123123
    
13. 1095591332aa----a111111
    
14. 1095591332aa----abc123
    
15. 1095591332aa----aini1314
    
16. 1095591332aa----iloveyou
    
17. 1095591332aa----q1w2e3r4
    
18. 1095591332aa----qq123123
    
19. 1111qqqq1111----a111111
    
20. 1111qqqq1111----abc123
    
21. 1111qqqq1111----aini1314
    
22. 1111qqqq1111----iloveyou
    
23. 1111qqqq1111----q1w2e3r4
    
24. 1111qqqq1111----qq123123
    
25. 111qqq111----a111111
    
26. 111qqq111----abc123
    
27. 111qqq111----aini1314
    
28. 111qqq111----iloveyou
    
29. 111qqq111----q1w2e3r4
    
30. 111qqq111----qq123123

复制代码

结果：

1. 1044122462aa---a111111
   
2. 1044122462aa---abc123
   
3. 1044122462aa---aini1314
   
4. 1005495524---a111111
   
5. 1005495524---abc123
   
6. 1005495524---aini1314
   
7. 1111qqqq1111---a111111
   
8. 1111qqqq1111---abc123
   
9. 1111qqqq1111---aini1314
   
10. 111qqq111---a111111
    
11. 111qqq111---abc123
    
12. 111qqq111---aini1314
    
13. 1095591332aa---a111111
    
14. 1095591332aa---abc123
    
15. 1095591332aa---aini1314
    
16. 1044122462aa---iloveyou
    
17. 1044122462aa---q1w2e3r4
    
18. 1044122462aa---qq123123
    
19. 1005495524---iloveyou
    
20. 1005495524---q1w2e3r4
    
21. 1005495524---qq123123
    
22. 1111qqqq1111---iloveyou
    
23. 1111qqqq1111---q1w2e3r4
    
24. 1111qqqq1111---qq123123
    
25. 111qqq111---iloveyou
    
26. 111qqq111---q1w2e3r4
    
27. 111qqq111---qq123123
    
28. 1095591332aa---iloveyou
    
29. 1095591332aa---q1w2e3r4
    
30. 1095591332aa---qq123123
    
31. 
    
32. 请按任意键继续. . .
    
33. 
    

复制代码

代码：

1. #!/usr/bin/env python3
   
2. #coding=utf-8
   
3. 
   
4. #用户名
   
5. userinfo = {}
   
6. with open('1.txt','r') as f:
   
7.     for each_line in f:
   
8.         temp = (each_line.split('----'))
   
9.         #去掉换行符
   
10.         if '\n' in temp[1]: temp[1] = temp[1][:-1]
    
11.         #用户名在字典中，追加数据
    
12.         if temp[0] in userinfo:
    
13.             userinfo[temp[0]].append(temp[1])
    
14.         else:
    
15.             userinfo[temp[0]] = [temp[1]]
    
16. 
    
17. #输出条件
    
18. limit = 3
    
19. #需要处理的用户
    
20. reuser = set(userinfo.keys())
    
21. rem_usr = set()
    
22. #不满足条件的用户
    
23. rem_usr_ = set()
    
24. 
    
25. # print(reuser)
    
26. while True:
    
27.     #减少不必要访问
    
28.     temp = reuser - rem_usr
    
29.     # print(temp)
    
30.     if temp == set(): break
    
31.     for usr in temp:
    
32.         temp_list = userinfo[usr]
    
33.         temp_len = len(temp_list)
    
34.         #过滤不满足条件的用户
    
35.         if temp_len < limit: rem_usr.add(usr)
    
36.         elif temp_len >= limit:
    
37.             for count in range(limit):
    
38.                 print(usr + '---' + temp_list[0])
    
39.                 temp_list.pop(0)
    
40.         #为0添加到剩余集合
    
41.         elif temp_len == 0: rem_usr_.add(usr)
    
42. 
    
43. temp = reuser - rem_usr_
    
44. #输出不满足条件的
    
45. if temp != set():
    
46.     for usr in reuser - rem_usr_:
    
47.         for u in userinfo[usr]:
    
48.             print(usr + '---' + u)

复制代码