2016-08-12 4 views
-2

I는 다음과 같습니다 CSV 파일이 있습니다. "러시아"조건부로 CSV 파일의 행을 추출 하시겠습니까?

Germany,1928,Food 
Iceland,1943,Oil 
France,1923,Plastics 
Russia,1901,Steal 
South Africa,1932,Silver 
Russia,1905,Gold 
Brazil,1901,Platinum 

나는 그것이 첫 번째 열을 검색하고 단어를 안타 행을 당겨 할을

이 내 코드는 현재 모습입니다 :

import csv 
import sys 

with open('country.csv', 'rb') as csvfile: 
    data = csv.reader(csvfile) 
    datalist = [] 
    for row in data: 
     if len (row) != 0: 
      datalist = datalist + [row] 
csvfile.close() 

column_names = datalist[0] # LIST OF COLUMNS 

나는 전체 행을 당기에 대해 어떻게 갈 것인가?

+0

귀하의 질문이 다소 불분명하다. "전체 행을 잡아 당기다"는 것은 무엇을 의미합니까? 전체 행을 목록에 추가하고 있습니다. 당신이 정말로하고 싶은 것을 설명하십시오. 또한, 귀하의 코드는 귀하의 설명에 명시된 바와 같이 어떤 필드도 '러시아'와 비교하지 않습니다. –

+0

주제 끄기 : 파일을 열 때'with open ... '을 사용하면 파일을 닫을 때 그 파일을 닫으면 자동으로 닫힙니다. – martineau

답변

1

filter을 시도해보십시오

>>> filter(lambda x: x[0] == 'Russia', datalist) 
[['Russia', '1901', 'Steal'], ['Russia', '1905', 'Gold']] 
+0

그게 하나야! 건배 남자! – semiflex

관련 문제