如何在python中查找csv文件中的特定行

Question

我想要做的是讀入csv文檔，並在SN列> 20中找到所有值，並僅創建一個新文件，其中SN> 20。

我知道，我需要做的：

1. 讀取原始文件
2. 打開了原始文件的行的新文件
3. 迭代

我已經能夠做的就是找到具有SN的值的行> 20

import csv 
import os 

os.chdir("C:\Users\Robert\Documents\qwe") 

with open("gdweights_feh_robert_cmr.csv",'rb') as f: 
    reader = csv.reader(f, delimiter= ',') 
    zerovar = 0 
    for row in reader: 
     if zerovar==0: 
      zerovar = zerovar + 1 
     else: 
      sn = row [11] 
      zerovar = zerovar + 1 
      x = float(sn) 
      if x > 20: 
       print x 

所以我的問題是我怎麼走行與SN> 20，並把它變成一個新的文件？

Answer 1

將數據保存在列表中，然後將列表寫入文件。

import csv 
import os 

os.chdir(r"C:\Users\Robert\Documents\qwe") 

output_ary = [] 
with open("gdweights_feh_robert_cmr.csv",'rb') as f: 
    reader = csv.reader(f, delimiter= ',') 
    zerovar = 0 
    for row in reader: 
     if zerovar==0: 
      zerovar = zerovar + 1 
     else: 
      sn = row [11] 
      zerovar = zerovar + 1 
      x = float(sn) 
      if x > 20: 
       print x 
       output_ary.append(row) 

with open("output.csv",'w') as f2: 
    for row in output_ary: 
     for item in row: 
      f2.write(item + ",") 

Answer 2

在代碼中，讀/循環遍歷行非常複雜。它可以被清理（和運行在Python更快）以下各項：

with open('gdweights_feh_robert_cmr.csv', 'rb') as f: 
    output_ary = [row for row in f if float(row[11]) > 20] 

使用列表理解（[一行行，如果F]）在蟒蛇優化，因此它會更有效地瓶坯。 AND ...避免必須創建reader數組，這將減少所需的內存，如果csv文件很大，也非常方便。

然後，您可以按照其他答案中的建議繼續寫出outout_ary。

希望這有助於！