Python-比较来自两个不同csv的两列中的相似值

Question

I have two CSV files with the same row names:

Name, Lastname.

However, file2.csv has an extra column named

Attention

Each file has a different list of name and lastname (not in order). I'm trying to find a way to find to print the attention column, if the name and lastname are in both files.

This is what I have so far:

    with open('result.csv') as r:
        set1 = set(x[0] for x in csv.reader(r))

    with open('result2.csv') as r:
        set2 = set(x[0] for x in csv.reader(r))

    for x, y in zip(set1, set2):
        if x[0] == y[0]:
            print("Matched")

How can I read the first and second column for each file?

Thank you

edit: being more clear

Answer 1

Create a set of tuples for the lookup table which will contain the values from your first two columns, eg:

with open("result.csv", "r") as f:
    result = {(x[0], x[1]) for x in csv.reader(f)}

And then just iterate over the second file, check if the tuple of the first two columns exists in the lookup table from the first file, and if yes - print the third column on each match, eg:

with open("result2.csv", "r") as f:
    for row in csv.reader(f):
        if (row[0], row[1]) in result:
            print("Matched: {}".format(row[2]))  # print the third column