2013-05-20 47 views
1

是它可以讀取這樣的CSV:讀CSV與一個命令的列表,然後在它讀

1,22,5 
1,33,9 
3,44,5 
4,32,5 
5,53,5 
6,64,5 
7,12,7 

用一個命令與格式[(1,22,5),...]列表?

如何在列表中搜索後得到所有元素第一列是1?

例如:與ROW1 == 1得到所有將返回

1,22,5 
1,33,9 

回答

2
import csv 

with open('data.csv', 'rb') as f: 
    rows = [[int(el) for el in row] for row in csv.reader(f)] 


>>> rows 
[[1, 22, 5], [1, 33, 9], [3, 44, 5], [4, 32, 5], [5, 53, 5], [6, 64, 5], [7, 12, 7]] 
>>> [row for row in rows if row[0] == 1] 
[[1, 22, 5], [1, 33, 9]] 
相關問題