在另一个文件中编写与字符匹配的行

Question

本质上，我想要编写与代码中引用的id列表相匹配的文档行。

nodeIDs.txt：

...有417个对象，

10000 
10023 
1017 
1019 
1021 
1026 
1027 
1029 
... 

粘着junction.txt：

...有73行，

的

4301: AFDN; afadin, adherens junction formation factor 
1496: CTNNA2; catenin alpha 2 
283106: CSNK2A3; casein kinase 2 alpha 3 
2241: FER; FER tyrosine kinase 
60: ACTB; actin beta 
1956: EGFR; epidermal growth factor receptor 
56288: PARD3; par-3 family cell polarity regulator 
10458: BAIAP2; BAI1 associated protein 2 
51176: LEF1; lymphoid enhancer binding factor 1 

我试图让程序一行一行并引用ids列表，并且如果行的开头字符与列表中找到的任何字符匹配，则将该行写入新文档。我正在研究数据集，但我不确定这些数据是否可用。

到目前为止我的代码：

ids = [] 
with open('nodeIDs.txt', 'r') as n: 
    for line in n: 
     ids.append(line) 
n.close() 

# Import data from the pathway file and turn into a list 
g = [] 
with open('Adherens junction.txt', 'r') as a: 
    for line in a: 
     g.append(line) 
a.close() 

aj = open('Adherens.txt', 'a') 
for line in a: 
    if ids[i] in line: 
    aj.write(line) 
aj.close() 

你能帮我得到这个工作？

Answer 1

这里有一些代码，我认为你做了什么之后。

代码：

# read ids file into a set 
with open('file1', 'r') as f: 
    # create a set comprehension 
    ids = {line.strip() for line in f} 

# read the pathway file and turn into a list 
with open('file2', 'r') as f: 
    # create a list comprehension 
    pathways = [line for line in f] 

# output matching lines 
with open('file3', 'a') as f: 

    # loop through each of the pathways 
    for pathway in pathways: 

     # get the number in front of the ':' 
     start_of_line = pathway.split(':', 1)[0] 

     # if this is in 'ids' output the line 
     if start_of_line.strip() in ids: 
      f.write(pathway) 

结果：

2241: FER; FER tyrosine kinase 
56288: PARD3; par-3 family cell polarity regulator 

file1的：

10000 
56288 
2241 

文件2：

4301: AFDN; afadin, adherens junction formation factor 
1496: CTNNA2; catenin alpha 2 
283106: CSNK2A3; casein kinase 2 alpha 3 
2241: FER; FER tyrosine kinase 
60: ACTB; actin beta 
1956: EGFR; epidermal growth factor receptor 
56288: PARD3; par-3 family cell polarity regulator 
10458: BAIAP2; BAI1 associated protein 2 
51176: LEF1; lymphoid enhancer binding factor 1 

什么是一套理解？

此：

# create a set comprehension 
ids = {line.strip() for line in f} 

相同：

# create a set 
ids = set() 
for line in f: 
    ids.add(line.strip())