python - 我想获取一个字符串列表，在另一个 csv 文件中找到它们并从匹配行打印一个字段

Question

我有一个很大的 .csv 文件，其中有几千行和几列。为了简单起见，它看起来有点像

Name, ID
One, 1
Two, 2

我想编写一个脚本，它接受一个字符串列表，将每一行视为一个单独的实体，并查找包含与列表中该条目匹配的行。一旦找到匹配的行，我想将其中一个字段打印到新文件中。

例如，给定一个列表:

1
2

我希望能够生成一个包含关联“名称”值的文件。

到目前为止，我在读取列表文件时没有遇到任何问题，例如:

list_file = sys.argv[1]
f = open(list_file, 'r+')
ListOfNames = f.read()

但我对如何继续感到有点困惑。以前，当我编写类似的代码时，我会使用 raw_input() 提示用户输入字符串，然后简单地在 csv 文件中搜索匹配项，如下所示:

 def Return_Name(ID):
        file_csv = csv.DictReader(open('file.csv'))
            for row in file_csv:
                if row['ID'] == ID:
                     print row['Name']

但我不确定如何在迭代列表时执行此操作。有人可以帮忙吗？

抱歉，这个问题不是最结构化的，但我们将不胜感激。

干杯

Answer 1

要读取 ID 列表，这应该可以完美实现:

list_ids = [line.strip() for line in open('list.txt').xreadlines()]

并将其与您发布的内容相结合:

for id in list_ids:
    Return_Name(id)

干杯