Условно извлекать строки CSV-файлов?

Question

У меня есть файл 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 

Как бы я идти о потянув всю строку?

Answer 1

filter Попробуйте:

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