放置条件后如何从csv文件获取标头

时间:2019-01-16 11:14:47

标签: python pandas csv

我正在寻找一种在将条件放入CSV行之后获取文件的列名的方法。

我得到了所需的输出,但没有得到列名,我必须在excel中获得带有列名或行标题的输出。

include_cols = ['TICKETID', 'TICKETNAME', 'TICKETSENTDATE', 
                'TICKETREQUESTER', 'TICKETCATEGORYNAME',
                'TICKETLOCATION', 'SERVICEGROUPNAME', 
                'SERVICETYPENAME', 'SERVICESTATENAME', 'REACTIONTIME', 
                'SOLUTIONTIME']

search_for = ['Resolved','Closed','Rejected']

with open(files3, "rt", encoding='utf-8') as f:
    reader = csv.DictReader(f, delimiter=';')
    for row in reader:
        if '+' in row['SOLUTIONTIME'] and row['SERVICESTATENAME'] not in search_for:
            content = list(row[i] for i in include_cols)
            print(content)

1 个答案:

答案 0 :(得分:0)

为什么不使用https://pandas.pydata.org/pandas-docs/stable/generated/pandas.read_csv.html将csv加载到熊猫数据框中?这样可以更轻松地检索列的名称。