Найти соответствующие столбцы между двумя CSV файлов

Question

у меня есть два файла CSV, содержащие следующие строки заголовка:

• файл 1:

  Product invoice,product verification,product not completed,product completed 
  

• Файл 2:

  Product invoice,product completed 
  

Мне нужно найти общие столбцы:

Product invoice,product completed 

Обратите внимание, что столбцы должны отображаться в том же порядке как в файлах1, так и в файле2.

Answer 1

Вы

list1 = ["Product invoice", "product verification", "product completed"] 
list2 = ["Product invoice", "product completed"] 

И хочу сравнить, если элементы list2 в list1, верно? Так

def compare(list1, list2): 
    list3 = [] 
    for elem in list1: 
     if elem in list2: 
      list3.append(elem) 
    return list3==list2 

Алгоритм состоит в создании List3 с элементами из list1, которые находятся в list2, а затем сравнивая, если результирующий список равен list2.

Это будет только True, если они находятся в том же порядке. Также учтите, что «Продукт завершен» отличается от «продукта завершен», поэтому вы можете захотеть, чтобы все строки были опущены (через str.lower()).

Answer 2

Сделать набор строк из каждой row[0] и row[2] в file1 форматирования, как именно он будет появляться в file2 перебирать не file2 увидеть, если появится строка в наборе:

import csv 
with open(file1) as f1, open(file2) as f2: 
    # skip headers 
    next(f1),next(f2) 
    r1 = csv.reader(f1) 
    # make set of strings matching format of file2 
    st = set("{},{}".format(row[0], row[2]) for row in r1) 
    # iterate over every line in file2 
    # and check if the line appears in the set 
    for line in f2: 
     if line.rstrip() in st: 
      print(line) 

File1:

Product invoice,product verification,product completed 
foo,1,2 
bar,3,4 
foo,foo,bar 

File2:

Product invoice,product completed 
foo,2 
bar,4 
foobar,foo 
bar,bar 

Выход:

foo,2 
bar,4 

Если вы хотите, чтобы данные в списках:

import csv 
with open(file1) as f1, open(file2) as f2: 
    r1 = csv.reader(f1) 
    r2 = csv.reader(f2) 
    st = set((row[0], row[2]) for row in r1) 
    for row in r2: 
     if tuple(row) in st: 
      print(row) 
['Product invoice', 'product completed'] 
['foo', '2'] 
['bar', '4']