Python CSV搜索脚本

Question

I wish to write a Python script which reads from a csv. The csv comprises of 2 columns. I want the script to read through the first column row by row and find the corresponding value in the second row. If it finds the value in the second row I want it to input a value into a third column.

example of output

Any help with this would be much appreciated and I hope my aim is clear. Apologies in advance if it is too vague.

Answer 1

this script read test.csv file and parse it an write to OUTPUT.txt

f = open("test.csv","r")
d={}
s={}
for line in f:
        l=line.split(",")
        if not l[0] in d:
                d[l[0]]=l[1].rstrip()   
                s[l[0]]=''
        else:
                s[l[0]]+=str(";")+str(l[1].rstrip())

w=open("OUTPUT.txt","w")
w.write("%-10s %-10s %-10s\r\n" % ("ID","PARENTID","Atachment"))
for i in d.keys():
        w.write("%-10s %-10s %-10s\r\n" % (i,d[i],s[i]))

f.close()
w.close()

---

example:

input:

ID         PARENTID   Atachment 
000001     sasa       ;sasa2;sas4
1023265    333                  
1          123        ;333;asas;333;ss
3                               
2          456                  
0221212  

OUTPUT:

 ID PARENTID Atachment 000001 sasa ;sasa2;sas4 1023265 333 1 123 ;333;asas;333;ss 3 2 456 0221212