有条件地提取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']]