python文档的数据读取，把读取数据写入到新的表里

目的：接口自动化过程需要从表格文件读取，然后把结果写到表格中。没有多余内容全部是精华！

读取文件1

读取文件2

 

 代码如下图：

# -*-coding:utf-8 -*-
# Author:wangjunqi
import csv

def read_csv_demo2():
    #result_file = "./result.csv"
    with open("china_smoking.csv","r") as fp:
        #使用dicreader创建的reader是一个字典对象，遍历后，不包含第一行数据
        results =[]
        reader = csv.DictReader(fp)
        for x in reader:
            pams = {}
            x = dict(x)
            location = x.get("location").strip()
            smoking = x.get("smoking").strip()
            smoking_yes = x.get("smoking_yes").strip()
            smoking_no = x.get("smoking_no").strip()
            pams["location"] =location
            pams["smoking"] = smoking
            pams["smoking_yes"] = smoking_yes
            pams["smoking_no"] = smoking_no
            pams["result"] = "通过"
            #print('测试字典形式读取：',pams)
            results.append(pams)
        print(results)
        with open ("result.csv","w",newline='') as csvfile:
            headers = "location,smoking,smoking_yes,smoking_no,result".split(",")
            writer = csv.DictWriter(csvfile, fieldnames=headers)
            ok = writer.writeheader()
            print(ok)
            for rt in results:
                writer.writerow(rt)
            csvfile.close()

read_csv_demo2()