csv 문서에서 파이썬으로 특정 행을 찾는 방법

Question

내가하려는 것은 CSV 문서를 읽고 SN 열> 20에서 모든 값을 찾고 SN> 20 인 행만 사용하여 새 파일을 만듭니다.

내가 할 필요가 있음을 알고

1. 원본 파일
2. 열기 원본 파일의 행을 통해 새 파일
3. 으로 반복 읽기

,691 내가 할 수있었습니다 무엇

는 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

코드에서 행을 통한 읽기/반복은 매우 복잡합니다. 그것은 청소 (빠른 파이썬에서 실행) 다음과 수 :

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

파이썬에 최적화되어 더 효율적으로 미리 형성 할 수 있도록, ([f는 경우 행에 대해 행]) 지능형리스트를 사용. 그리고 ... reader 배열을 만들 필요가 없기 때문에 필요한 메모리가 줄어들며 csv 파일이 큰 경우 매우 편리합니다.

그러면 다른 답변에서 제안한대로 outout_ary를 작성할 수 있습니다.

희망이 도움이됩니다.