如何在 csv 文件中搜索特定字段，然后在 Python 中打印整行

Question

我会包含我的整个程序，但它太长了，所以我将提供一个示例：

假设我有这个 csv 文件：

John, 23, 1944, Africa
Fred, 45, 1922, China
Bob, 23, 1999, Japan

如何搜索文件并打印 1944 所在的行？ 基本上我想要它做的是找到 1944，然后打印 1944 所在的整行。

另外，如果它的结构类似于

Name, Age, Year, Nation
John, 23, 1944, Africa
Fred, 45, 1922, China
Bob, 23, 1999, Japan

您将如何打印找到的行以及顶行？

非常感谢。

编辑： 为什么需要反对投票？我合法地找不到这方面的任何信息。所以我不得不问。

Answer 1

这个程序回答了你的第一个问题：

import csv
import sys

with open('csv1.txt') as fp:
    reader = csv.reader(fp, skipinitialspace=True)
    writer = csv.writer(sys.stdout)
    writer.writerows(row for row in reader if row[2] == '1944')

这回答了你的第二个问题：

import csv
import sys

with open('csv2.txt') as fp:
    top_row = next(fp)
    sys.stdout.write(top_row)
    reader = csv.reader(fp, skipinitialspace=True)
    writer = csv.writer(sys.stdout)
    writer.writerows(row for row in reader if row[2] == '1944')