Python：如何处理 CSV 中的缺失值？答案

【问题标题】：Python: How to handle missing values in a CSV?Python：如何处理 CSV 中的缺失值？
【发布时间】：2016-07-04 19:01:12
【问题描述】：

我有一个给定的 CSV 样本，如下所示：

ID,ID_TYPE,OB_DATE,VERSION_NUM,MET_DOMAIN_NAME,OB_END_CTIME,OB_DAY_CNT,SRC_ID,REC_ST_IND,PRCP_AMT,OB_DAY_CNT_Q,PRCP_AMT_Q,METO_STMP_TIME,MIDAS_STMP_ETIME,PRCP_AMT_J
90, RAIN, 2006-01-01 00:00,1, WADRAIN,900,1,24109,1011,0,0,6, 2006-01-17 09:04,0,
150, RAIN, 2006-01-01 00:00,1, DLY3208,900,1,30747,1011,0,0,6, 2006-01-09 13:21,3,
174, RAIN, 2006-01-01 00:00,1, WADRAIN,900,1,24775,1011,0.2,0,6, 2006-01-17 09:04,0,

我想在我的 CSV 中确定每个给定日期的工作日。我的代码实现如下：

import csv
from datetime import datetime as dt


csv_file = open('raindata.csv')
csv_reader = csv.DictReader(csv_file)
field_names = list(csv_reader.fieldnames)
if 'WEEKDAY' in field_names:
    print "data has error"

elif 'RECWEEKDAY' in field_names:
    print "data has error"

else:
    field_names.insert(field_names.index('OB_DATE') + 1, 'WEEKDAY')
    field_names.insert(field_names.index('METO_STMP_TIME') + 1, 'RECWEEKDAY')

    def get_weekday(ob_date):
        return dt.strptime(ob_date, ' %Y-%m-%d %H:%M').strftime('%A')

    output = open('raindata.csv','w')
    csv_writer = csv.DictWriter(output, field_names)
    csv_writer.writeheader()
    for row in csv_reader:
        row['WEEKDAY'] = get_weekday(row['OB_DATE'])
        row['RECWEEKDAY'] = get_weekday(row['METO_STMP_TIME'])
        csv_writer.writerow(row)

我的脚本运行良好并给出了正确的结果，但是如果 OB_DATE 列和 METO_STMP_TIME 列中缺少 Date 值，它就会失败。

如何更改现有代码，以便对于空白的Date 值，对应的Weekday 值也为空白？

【问题讨论】：

@Rahul : ValueError: 时间数据“”与格式“%Y-%m-%d %H:%M”不匹配
@Rahul 我知道它采用特定格式的字符串，但我希望对于空白它应该给出空白。

标签： python csv python-2.x

【解决方案1】：

对于其他选择，您可以修改 get_weekday 函数以处理空白日期。

def get_weekday(ob_date):
    return dt.strptime(ob_date, ' %Y-%m-%d %H:%M').strftime('%A') if ob_date.strip() else ""

【讨论】：

【解决方案2】：

只需捕获日期/时间字符串丢失或无效时引发的异常，然后将值设置为空字符串。

try:
    row['WEEKDAY'] = get_weekday(row['OB_DATE'])
except ValueError:
    row['WEEKDAY'] = ''

【讨论】：

也可以使用if row["OB_DATE"]: ... else:...而不是try:... except:...，但这完全取决于偏好。
输出什么都没有。 CSV 为空白
@TadhgMcDonald-Jensen：try:...except:... 也适用于无效日期（或格式错误的日期）。因此，这可能是比专门检查虚假值更好的解决方案。
@jsfan，如果日期格式错误，我希望程序失败，就像我说的那样：这真的取决于偏好。
@desmond.carros：如果你有足够的内存，你可以先把整个文件读入内存，然后直接写回文件。