Python CSV DictReader 忽略列？

Question

如果我使用 CSV.dictReader 读取 CSV，我该如何让它忽略 CSV 中的某些列？

例如，

"id","name","address","number","created"
"123456","someName","someAddress","someNumber","2003-5-0294"

我只想使用读取器获取 ID 和名称，丢弃并忽略其余部分。我尝试使用字段名，但它仍然读取它并将其设置为“无”。我注意到 csv.DictWriter 有一个“忽略”功能，但 DictReader 似乎没有。希望有一种更优雅的方式来做到这一点，而不是仅仅读取然后只将我想要的列写入另一个 CSV，然后使用 DictReader 读取该 CSV 以进行进一步处理。

谢谢大家!

Answer 1

这个简单的生成器就可以做到。

def dict_filter(it, *keys):
    for d in it:
        yield dict((k, d[k]) for k in keys)

像这样使用它:

dreader = [{'id':1, 'name':'Bob', 'other_stuff':'xy'},
           {'id':2, 'name':'Jen', 'other_stuff':'xx'}]

for d in dict_filter(dreader, 'id', 'name'):
    print d

给出:

{'id': 1, 'name': 'Bob'}
{'id': 2, 'name': 'Jen'}