Python實現大資料收集至excel的思路詳解
阿新 • • 發佈:2020-01-09
一、在工程目錄中新建一個excel檔案
二、使用python指令碼程式將目標excel檔案中的列頭寫入,本文省略該部分的code展示,可自行網上查詢
三、以下code內容為:實現從介面獲取到的資料值寫入excel的整體步驟
1、整體思路:
(1)、根據每日調取介面的日期來作為excel檔案中:列名為“收集日期”的值
(2)、程式預設是每天會定時調取介面並獲取介面的返回值並寫入excel中(我使用的定時任務是:linux下的contab)
(3)、針對介面異常未正確返回資料時,使用特殊符號如:NA代替並寫入excel檔案中(後期使用excel資料做分析時有用)
2、完整程式碼如下:
import requests,xlrd,os,sys,urllib3 from datetime import date,timedelta from xlutils.copy import copy basedir = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) sys.path.append(basedir) from lib.mysqldb import mysqldb from lib.public_methods import test_login def collect_data(): """test_rooms.test_kpi卡片下:adr指標值收集""" get_all_code_sql = 'select DISTINCT test_code from test_info WHERE open_flag = 1' test_code_all = mysqldb("test_data").selectsql(get_all_code_sql) test_code_list = [] adr_insert_data_list = [] yesterday = (date.today() + timedelta(days=-1)).strftime("%Y-%m-%d") adr_insert_data_list.append(yesterday) for j in range(len(test_code_all)): test_code_list.append(test_code_all[j]["test_code"]) for m in range(len(test_code_list)): url = "https://www.baidu.com/test/api/data/query.json" header = { "Content-Type": "application/json;charset=UTF-8","Cookie": str(test_login()) } param = { "code": "test_rooms.test_kpi","page": 1,"pageSize": 1000,"params": { "start_date_year": "2019","start_date_month": "9","start_date_day": "16","end_date_year": "2019","currency_type": "usd","end_date_day": "16","end_date_month": "9","tests": "test_001" } } """替換請求引數中的開始日期""" param["params"]["start_date_year"] = str(yesterday).split("-")[0] param["params"]["start_date_month"] = str(yesterday).split("-")[1] param["params"]["start_date_day"] = str(yesterday).split("-")[2] """替換請求引數中的結束日期""" param["params"]["end_date_year"] = param["params"]["start_date_year"] param["params"]["end_date_month"] = param["params"]["start_date_month"] param["params"]["end_date_day"] = param["params"]["start_date_day"] param["params"]["tests"] = test_code_list[m] urllib3.disable_warnings() result = requests.post(url=url,headers=header,json=param,verify=False).json() if str(result["data"]["data"]) != "None": """adr指標值收集""" indicatorList = result["data"]["data"]["test_indicator_list"] test_actualorLast_Forecast = result["data"]["data"]["test_actual"] new_indicator_actualvalue = {} i = 0 while i < len(indicatorList): dit = {indicatorList[i]: test_actualorLast_Forecast[i]} new_indicator_actualvalue.update(dit) i += 1 if str(new_indicator_actualvalue["adr"]) == "--": adr_value_result = "NA" adr_insert_data_list.append(adr_value_result) else: adr_value_result = new_indicator_actualvalue["adr"] adr_insert_data_list.append(adr_value_result) else: adr_value_result = "NA" adr_insert_data_list.append(adr_value_result) """adr指標值資料收集入excel路徑""" workbook = xlrd.open_workbook(basedir + "/data/collect_data_center.xls") # 開啟工作簿 sheets = workbook.sheet_names() # 獲取工作簿中的所有表格 worksheet = workbook.sheet_by_name(sheets[0]) # 獲取工作簿中所有表格中的的第一個表格 rows_old = worksheet.nrows # 獲取表格中已存在的資料的行數 new_workbook = copy(workbook) # 將xlrd物件拷貝轉化為xlwt物件 new_worksheet = new_workbook.get_sheet(0) # 獲取轉化後工作簿中的第一個表格 for i in range(0,1): for j in range(0,len([adr_insert_data_list][i])): new_worksheet.write(i + rows_old,j,[adr_insert_data_list][i][j]) # 追加寫入資料,注意是從i+rows_old行開始寫入 new_workbook.save(basedir + "/data/collect_data_center.xls") # 儲存工作簿 print("adr指標值---xls格式表格【追加】寫入資料成功!")
3、從步驟2中的程式碼可看出程式碼整體分為3個部分:
(1)、組裝介面引數;
(2)、呼叫介面將介面返回的結果集收集在list中;
(3)、將收集的結果寫入excel中並儲存;
tips:windows與linux下excel的路徑格式需要區分下,以上程式碼中的"/data/collect_data_center.xls"為linux環境下路徑
總結
以上所述是小編給大家介紹的Python實現大資料收集至excel的思路詳解,希望對大家有所幫助,如果大家有任何疑問請給我留言,小編會及時回覆大家的。在此也非常感謝大家對我們網站的支援!