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
是它可以讀取這樣的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
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]]