python按照某列把数据提出 python根据a列提取b列的数据

标题有点拗口，但做过数据库编程可能比较容易理解，就是查询一张表的主键，然后再另外一张表中去找包含这个主键的行，把该行信息保存下来。今天晚上在看上午的测试结果，想把数据可视化分析一下算法的问题。但是发现测试程序保存的数据不完整。本来心一横，想自己一行一行处理，但是发现自己是程序猿啊，能不能通过python脚本处理一下，果然很简单。

alines = [str(l.split()[0]) for l in open("a.txt")]
fw = open('c.txt','w')
for al in alines:
    for line in open('b.txt','r'):
        if al in line:
            fw.write(line)
fw.close()

a.txt示例如下：

b.txt示例如下：

 

c.txt示例如下：