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

Question

I would include my whole program but it is too long so I will provide an example:

Say I had this csv file:

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

How do I search the file and print the row that 1944 is specifically in? Essentially what I want it to do is find 1944, then print the whole row 1944 is in.

Also, if it's structured like

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

How would you print the found row plus the top row as well?

Many thanks.

Edit: Why is down voting necessary? I legitimately cannot find any information on this. Therefore I have to ask.

Answer 1

This program answers your first question:

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')

And this answers your second question:

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')