Python - Compare similar values in two columns from two different 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

@Neo My bad, meant to say compare first and second column of each file. — Blinhawk
– Blinhawk, Commented Jul 23, 2018 at 13:41

zwer · Accepted Answer · 2018-07-23 13:57:18Z

2

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

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, e.g.:

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

edited Jul 23, 2018 at 13:57

answered Jul 23, 2018 at 13:45

zwer

25.9k3 gold badges53 silver badges70 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Python - Compare similar values in two columns from two different csv

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related