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)