编辑 CSV 文件中的查询

Question

我在列表中有列名，我想将值 1 分配给这些列名，并将值 0 分配给其余列。 例如。

perms_name = ['name','place','thing']

最初的 CSV 看起来像：

name,age,place,thing,phone_no

我想让 csv 看起来像：

name,age,place,thing,phone_no
1,0,1,1,0

我可以通过这个简单地做到这一点

with open('eggs.csv','a') as csvfile:
    fieldname = ["name","age","place","thing","phone_no"]
    writer = csv.DictWriter(csvfile,fieldnames=fieldname)
    writer.writeheader()

    writer.writerow(
        {'name': 1,'age':0,'place':1,'thing':1,'phone_no':0}
    )

但是他们有更快的方法吗，在这种情况下，它们只有 5 列，如果它们是 100 列，我只想将 1 分配给列表中提到的列。

Answer 1

您可以通过** 运算符在 Python 3 中解压字典。结合dict.fromkeys，无需为每个字段显式写出键值：

perms_name = ['name','place','thing']
fieldname = ['name', 'age', 'place', 'thing', 'phone_no']

d = {**dict.fromkeys(fieldname, 0), **dict.fromkeys(perms_name, 1)}

{'age': 0, 'name': 1, 'phone_no': 0, 'place': 1, 'thing': 1}

定义你的字典一次然后使用：

writer.writerow(d)

Answer 2

对于 jpp 的回答，您可以使用稍微令人困惑的列表推导来构建字典。

perms_name = ['name','place','thing']
fieldname = ['name', 'age', 'place', 'thing', 'phone_no']
d = {field : (1 if field in perms_name else 0) for field in fieldname}
# {'name': 1, 'age': 0, 'place': 1, 'thing': 1, 'phone_no': 0}

如果有任意数量的列从 csv 中读取它们而不是将它们硬编码到脚本中，这也可能是值得的。