python - 使用 DictWriter 编写字典键的子集

Question

我使用 csv 模块编写了一个将字典列表序列化为 CSV 文件的函数，代码如下:

data = csv.DictWriter(out_f, fieldnames)
data.writerows(dictrows)

但是，有时我只想将每个字典键的一个子集写到一个文件中。如果我将每个字典所具有的键的子集作为 fieldnames 传递，则会出现错误:

"dict contains fields not in fieldnames"

我怎样才能让 DictRows 只将我指定的字段的子集写入 CSV，而忽略那些在字典中但不在字段名中的字段？

Answer 1

最简单和最直接的方法是在初始化 DictWriter 实例时传递 extrasaction='ignore'，如记录 here :

If the dictionary passed to the writerow() method contains a key not found in fieldnames, the optional extrasaction parameter indicates what action to take. If it is set to 'raise' a ValueError is raised. If it is set to 'ignore', extra values in the dictionary are ignored.

它也适用于 writerows，它在内部只是重复调用 writerow。