python csv writer如果行键不存在

Question

以下脚本出错：

import csv,time,string,os,requests, datetime

test = "\\\\network\\Shared\\test.csv"

fields = ["id", "Expiration Date", "Cost", "Resale" ]

with open(test) as infile, open("c:\\upload\\tested.csv", "wb") as outfile:
    r = csv.DictReader(infile)
    w = csv.DictWriter(outfile, fields, extrasaction="ignore")
    r = (dict((k, v.strip()) for k, v in row.items() if v) for row in r)

    wtr = csv.writer( outfile )                                                                                                                                                                                                                                                                                                                                         
    wtr.writerow(["id", "upload_date", "cost", "resale"])
    for i, row in enumerate(r, start=1):
        row['id'] = i
        print(row['Expiration Date']
        row['Expiration Date'] = datetime.datetime.strptime(row['Expiration Date'][:10], "%m/%d/%Y").strftime("%Y-%m-%d")
        w.writerow(row)


D:\Python\Scripts>python test.py
Traceback (most recent call last):
  File "test.py", line 18, in <module>
    print(row['Expiration Date'])
KeyError: 'Expiration Date'

所以我想我明白发生了什么 - 来自原始文件的类似内容：

Expiration Date     Cost     Resale
2016-01-01          1.00     2.00
                    1.42     2.42
2016-05-02          1.45     9.00

据我所知，有一行没有填充到期日期列。如何强制 DictWriter 跳过空白 - 假设这是我的错误的原因？

Answer 1

实际上，csv.DictReader 生成的dict 只是将None 放入它找不到的字段中，因此您不应该收到该错误。您没有使用DictReader 的功能来生成正确的dict！据我所知，您尝试使用r = (dict((k, v.strip()) for k, v in row.items() if v) for row in r) 行自己进行解析。但是，这实际上不起作用。如果您之后打印这些行，您会得到：

{'Expiration Date     Cost     Resale': '2016-01-01          1.00     2.00'}
{'Expiration Date     Cost     Resale': '1.42     2.42'}
{'Expiration Date     Cost     Resale': '2016-05-02          1.45     9.00'}

所以每个dict 只包含一个密钥。您的文件的一个问题是，您在行之间没有有效的分隔符。看起来您的意思是使用空格，但Expiration Date 中也有空格。你将不得不摆脱它。如果你这样做，那么你可以像这样使用DictReader：

import csv,time,string,os,requests, datetime

test = "test.csv"

with open(test) as infile:
    r = csv.DictReader(infile, delimiter=" ", skipinitialspace=True)
for row in r:
    print(row)

现在给你：

{'Resale': '2.00', 'Cost': '1.00', 'ExpirationDate': '2016-01-01'}
{'Resale': None, 'Cost': '2.42', 'ExpirationDate': '1.42'}
{'Resale': '9.00', 'Cost': '1.45', 'ExpirationDate': '2016-05-02'}

这是一个正确的dict（请注意，读者无法知道第一个元素是缺失的元素）。现在您只需从写入中排除不完整的行。描述了一个很好的方法here：

import csv,time,string,os,requests, datetime

test = "test.csv"

with open(test) as infile:
    r = csv.DictReader(infile, delimiter=" ", skipinitialspace=True)

    for row in r:
        if not any(val in (None, "") for val in row.itervalues()):
            print(row)

最后，这将为您提供dicts 的所有有效行：

{'Resale': '2.00', 'Cost': '1.00', 'ExpirationDate': '2016-01-01'}
{'Resale': '9.00', 'Cost': '1.45', 'ExpirationDate': '2016-05-02'}

Answer 2

您在访问 x['Expiration Date'] 的字典中没有的内容时遇到 KeyError，因此您可以说 x.get('Expiration Date') 或可能是 'Expiration Date' in x 来发现它是否存在并有条件地丢弃该行。