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()