python 連接數據庫，查詢結果寫入數據到excel

##連接數據庫，查詢結果寫入數據到excel
import pymysql
import xlwt
def get_sel_excel(file_excel):
    #建立連接
    conn = pymysql.connect(
        host = "100.100.100.0",
        port = 3306,
        user = "test",
        passwd = "123",
        db = "bigdata",
        charset = "utf8"
    )

    #建立游標
    cursor = conn.cursor()
    sel_sql = "select * from ts_vehicle_load;"
    print("開始查詢表！")
    #執行sql語句
    cursor.execute(sel_sql)
    #獲取查詢到結果
    res = cursor.fetchall()
    print(res)
    w_excel(res)


#操作excel
def w_excel(res):
    book = xlwt.Workbook() #新建一個excel
    sheet = book.add_sheet('vehicle_land') #新建一個sheet頁
    title = ['id','vehicle_load_cod','total_count','vehicle_innet_count','vehicle_online_count','date']
    #寫表頭
    i = 0
    for header in title:
        sheet.write(0,i,header)
        i+=1


    #寫入數據
    for row in range(1,len(res)):
        for col in range(0,len(res[row])):
            sheet.write(row,col,res[row][col])
        row+=1
    col+=1
    book.save('vehicle_land.xls')
    print("導出成功！")


if __name__ == "__main__":
    file_excel = r"E:\Users\admin\PycharmProjects\untitled\1.xls"
    get_sel_excel(file_excel)