在csv文件中搜索列表Python中的元素

Question

I need some help with a script I made in Python and it's not working.

So I have a file with some identifiers (numbers), that I turn into a list. so far so good!

Then I want to check a .csv file, and if the first element of the line is on the previous list write that line to a new .csv file.

My script looks like this:

tratamento = open('therapies.csv')
tratados_B = open('tratados_SubB.csv')
Outfile = open('subtype_B_Tratamento.csv', 'w')

treated_list = []


for line in tratados_B:
    TL = line.strip()
    treated_list.append(TL) 

for line in tratamento:
    TLB = line.strip().split(';')[0]
    if TLB in treated_list:
        Outfile.write(line)
    else: 
        pass

Any idea of what am I doing wrong?

Thank you in advance!

Answer 1

Try to add Outfile.close() at the end of your script.

A better way of doing this is to use the with keyword.

treated_list = []

with open('subtype_B_Tratamento.csv', 'w') as Outfile,
     open('therapies.csv') as tratamento,
     open('tratados_SubB.csv') as tratados_B:

    for line in tratados_B:
        TL = line.strip()
        treated_list.append(TL) 

    for line in tratamento:
        TLB = line.strip().split(';')[0]
        if TLB in treated_list:
            Outfile.write(line)

With handles the closing of the file for you.