1. 程式人生 > 程式設計 >Python實現大資料收集至excel的思路詳解

Python實現大資料收集至excel的思路詳解

一、在工程目錄中新建一個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的思路詳解,希望對大家有所幫助,如果大家有任何疑問請給我留言,小編會及時回覆大家的。在此也非常感謝大家對我們網站的支援!

如果你覺得本文對你有幫助,歡迎轉載,煩請註明出處,謝謝!