该博客介绍了一个Python函数，用于从CSV文件中读取指定列，并将数据转换为namedtuple或tuple对象。示例代码展示了如何使用该函数从CSV数据中提取并打印特定列。
def read_csv(file, columns, type_name="Row"):
try:
row_type = namedtuple(type_name, columns)
except ValueError:
row_type = tuple
rows = iter(csv.reader(file))
header = rows.next()
mapping = [header.index(x) for x in columns]
for row in rows:
row = row_type(*[row[i] for i in mapping])
yield row
例：
>>> import csv
>>> from collections import namedtuple
>>> from StringIO import StringIO
>>> def read_csv(file, columns, type_name="Row"):
... try:
... row_type = namedtuple(type_name, columns)
... except ValueError:
... row_type = tuple
... rows = iter(csv.reader(file))
... header = rows.next()
... mapping = [header.index(x) for x in columns]
...
最低0.47元/天 解锁文章
5085
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
