请教一个在列表或者字典中取值相加的问题

雨久

有两个文件，想实现第一个文件所列出名字，在第二个文件中获取信息，如果第二个文件中没有记录则返回 no hit.
其中第一个文件如下：
Sam
Tom
Billy
Fang
Yong
Bill

另外一个文件如下：
Sam,25
Sam,teacher
Sam,America
Tom,student
Tom,ITer
Billy,38
Billy,professor
Fang,famer
Yong,Phd

根据上述两个文件得到如下结果文件：
Sam, 25;teacher;America
Tom, student;ITer
Billy, 38;professor
Fang, famer
Yong, Phd
Bill, no hit

本来想用建立字典然后遍历取结果，可是字典同样的键只能对应一个值，没能实现。后来想构建列表，也没有运行出来。在此请教各位兄弟姐妹。

icymirror

可以考虑使用字典中的值是列表的方式来组织数据。

timespace

回复 1# 雨久
印象中很久以前你也问过类似的问题，Python入门应该很快啊

1. import io
   
2. 
   
3. infile1 = io.StringIO(
   
4. '''Sam
   
5. Tom
   
6. Billy
   
7. Fang
   
8. Yong
   
9. Bill
   
10. ''')
    
11. 
    
12. infile2 = io.StringIO(
    
13. '''Sam,25
    
14. Sam,teacher
    
15. Sam,America
    
16. Tom,student
    
17. Tom,ITer
    
18. Billy,38
    
19. Billy,professor
    
20. Fang,famer
    
21. Yong,Phd
    
22. ''')
    
23. 
    
24. 
    
25. def main():
    
26.     name_db = {}
    
27.     for line in infile2:
    
28.         fields = line.rstrip().split(',')
    
29.         if len(fields) == 2:
    
30.             name_db.setdefault(fields[0], []).append(fields[1])
    
31.     for line in infile1:
    
32.         name = line.rstrip()
    
33.         value = ';'.join(name_db[name]) if name in name_db else 'no hit'
    
34.         print(name, value, sep=',')
    
35. 
    
36. if __name__ == '__main__':
    
37.     main()

复制代码

保存为test.py，执行：

1. bash-3.2 $python --version
   
2. Python 3.3.4
   
3. bash-3.2 $python test.py
   
4. Sam,25;teacher;America
   
5. Tom,student;ITer
   
6. Billy,38;professor
   
7. Fang,famer
   
8. Yong,Phd
   
9. Bill,no hit

复制代码

雨久

感谢timespace大神:wink::wink::wink:
平时没有努力学习啊，大神见笑了。
把代码改写如下，没有运行出来

import sys

def main():
    if len(sys.argv) != 4:
        sys.exit('usage: {} infile1 infile2 outfile'.format(sys.argv[0]))
   
    with open(sys.argv[1]) as in1, open(sys.argv[2]) as in2, open(sys.argv[3], 'w') as out:
        name_db = {}
        for line in in2:
            fields = line.rstrip().split(',')
            if len(fields) == 2:
                name_db.setdefault(fields[0], []).append(fields[1])
        for name in in1:
            name = name.rstrip()
            value = ';'.join(name_db[name]) if name in name_db else 'no hit'
            print >> out , name, ',',value

if __name__ == '__main__':
    main()

timespace

回复 4# 雨久
报什么错？贴出内容


   

雨久

没有报错，只是除了no hit，其它的值都没有写进去，结果如下：

Sam
Tom
Billy
Fang
Yong
Bill, no hit

timespace

回复 6# 雨久
很容易就改为2.7版本的：

1. from cStringIO import StringIO
   
2. 
   
3. infile1 = StringIO(
   
4. '''Sam
   
5. Tom
   
6. Billy
   
7. Fang
   
8. Yong
   
9. Bill
   
10. ''')
    
11. 
    
12. infile2 = StringIO(
    
13. '''Sam,25
    
14. Sam,teacher
    
15. Sam,America
    
16. Tom,student
    
17. Tom,ITer
    
18. Billy,38
    
19. Billy,professor
    
20. Fang,famer
    
21. Yong,Phd
    
22. ''')
    
23. 
    
24. def main():
    
25.     name_db = {}
    
26.     for line in infile2:
    
27.         fields = line.rstrip().split(',')
    
28.         if len(fields) == 2:
    
29.             name_db.setdefault(fields[0], []).append(fields[1])
    
30.     for line in infile1:
    
31.         name = line.rstrip()
    
32.         value = ';'.join(name_db[name]) if name in name_db else 'no hit'
    
33.         print '{},{}'.format(name, value)   
    
34. 
    
35. if __name__ == '__main__':
    
36.     main()

复制代码

运行：

1. bash-3.2 $python2.7 test.py
   
2. Sam,25;teacher;America
   
3. Tom,student;ITer
   
4. Billy,38;professor
   
5. Fang,famer
   
6. Yong,Phd
   
7. Bill,no hit

复制代码

如果参考这个，还得不出结果，坦白说你不适合用Python，换个方向吧，可能更有效率。

   

雨久

非常感谢timespace，您提供的代码很好用。能得到想要的结果，谢谢！