两个 csv 文件之间的相似之处

Question

我对 Python 还很陌生，并且在尝试制作这个查找两个 csv 文件之间匹配项的程序时遇到了很多麻烦。例如，我有两个 csv 文件。第一个称为“列表”，第二个称为“示例”

文件“列表”在第一行包含以下内容： 腿，膝盖，大腿，小腿，脚踝，臀部，脚，脚趾，小腿，脚，髌骨，胫骨，腓骨

文件“示例包含： 学生昨天摔断了腿，今天学生摔断了手臂，今天学生扭伤了大腿，学生扭肘，今天学生扭伤了脚踝

所以基本上如果 csv 文件“example”包含 csv 文件“list”中的任何单词，它应该将其输出到一个新的 csv 文件中，其中包含 example 中的句子，但不是。

到目前为止，这是我的代码：`

import csv

   with open("list.csv", "U") as file1, open("example.csv", "rb") as
   file2,open("finalOutput.csv", "wb") as outputfile:
   reader1 = csv.reader(file1,delimiter=';')
   reader2 = csv.reader(file2,delimiter='|')
   writer = csv.writer(outputfile,delimiter='|')

   rows2 = [row for row in reader2]
   for row1 in reader2:
       for row2 in rows2:
           if row1[0] == row2[0]:
               data = [row1[0],row2[0]]
               print data
               writer.writerow(data)

Answer 1

为什么不尝试这样的事情（假设如果任何单词与第二个文件中的单词匹配，您希望打印整行。基本上，您将第二行设为字符串，然后检查第一个文件中的任何单词是否在那个字符串。如果是，写出来。

with open("list.csv", "U") as file1, open("example.csv", "rb") as file2, open("output.csv", "wb+") as file3:
    reader1 = csv.reader(file1)
    reader2 = csv.reader(file2)
    writer = csv.writer(file3)

    reader1_rows = [row for row in reader1]
    reader2_rows = [row for row in reader2]

    for num, row in enumerate(reader1_rows):
        if ([word for word in row if word in ' '.join(reader2_rows[num])]):
            writer.writerow([row, reader2_rows[num]])

---

根据您调整后的评论，我相信这应该会为您提供所需的输出：

with open("list.csv", "U") as file1, open("example.csv", "rb") as file2, open("output.csv", "wb+") as file3:
    reader1 = csv.reader(file1)
    reader2 = csv.reader(file2)
    writer = csv.writer(file3)

    reader1_rows = [row for row in reader1]
    reader2_rows = [row for row in reader2]

    for num, row in enumerate(reader1_rows):
        for word in reader2_rows[num]:
            for item in row:
                if item in word:
                    writer.writerow([item, word])

更“pythonic”的方式可能如下：

with open("list.csv", "U") as file1, open("example.csv", "rb") as file2, open("output.csv", "wb+") as file3:
    reader1 = csv.reader(file1)
    reader2 = csv.reader(file2)
    writer = csv.writer(file3)

    reader1_rows = [row for row in reader1]
    reader2_rows = [row for row in reader2]

    for rowA, rowB in zip(reader1_rows, reader2_rows):
        for word in rowA:
            for item in (item for item in rowB if word in item):
                writer.writerow([word, item])

---

如果您想对齐列中的所有数据（为此，您可能应该这样做）并且数据如下所示：

leg
knee
thigh
shin
ankle
hip
foot
toe
calf
feet
patella
tibia
fibula

..和..

Student broke leg yesterday
Student broke arm today
Student hurt thigh today
Student twisted elbow
Student rolled ankle today

..那么你可以这样做：

with open("example.csv") as file1, open("list.csv") as file2, open("output.csv", "wb+") as file3:
    writer = csv.writer(file3)
    key_words = [word.strip() for word in file2.readlines()]
    for row in file1:
        row = row.strip()
        for key in (key for key in key_words if key in row):
            writer.writerow([key, row])

Answer 2

据我所知，你的 csv 文件的结构，我认为你不应该使用 csv-reader 来加载你的示例文件和你的话......

import csv

with open("list.csv", "U") as file1, open("example.csv", "rb") as
    file2,open("finalOutput.csv", "wb") as outputfile:

    writer = csv.writer(outputfile,delimiter='|')

    words = set(file1.read().split(','))

    # examples are split by "," so read the whole file and split it by ","
    examples = file2.read().split(',')

    for word in file1:
        for example in examples:
            # if the word happens to be within the example
            if word in example:
                   # add it to your output file
                   data = [word,example]
                   print data
                   writer.writerow(data)