python - 在 csv 中搜索值并多次返回该行

Question

我是一个尝试学习Python的菜鸟，

我正在尝试为包含 30,000 行数据的 CSV 文件编写脚本。

我想在每一行中查找列中的数字，并在每次找到该数字时返回该行。

我搜索并尝试了许多不同的建议，但他们似乎没有做我需要的事情，任何人都可以帮助我，如果我没有道理，请告诉我。

这是我到目前为止所拥有的，它仅返回标题:

    import csv

with open("test.csv", "r") as input, open ("result.txt","w") as result:
          testfilereader = csv.DictReader(input) 
          Age = 23
          fieldnames = testfilereader.fieldnames
          testfilewriter = csv.DictWriter(result, fieldnames, delimiter=',',)
          testfilewriter.writeheader()      
          for row in testfilereader:
                 for field in row:
                        if field == Age:
                             testfilewriter(row)


input.close

谢谢大家

Answer 1

您可以按如下方式使用Pandas:

csv 文件:

Id,Name,Age
1,John,30
2,Alex,20
3,Albert,30
4,Richard,30
5,Mariah,30

python :

import pandas as pd

df = pd.read_csv("ex.csv", sep = ",")
print df[df["Age"] == 30] 

   Id     Name  Age
0   1     John   30
2   3   Albert   30
3   4  Richard   30
4   5   Mariah   30