python笔记- excel操作

 前提：

python操作excel需要使用的模块有xlrd、xlwt、xlutils。对excel进行读、写、更新操作。操作excel时需要先导入这些模块，demo如下：

excel-读操作知识点：

import xlrd
'''
读取 excel的操作步骤如下：
1. 打开excel，打开的excel必须存在
2. 获取sheet对象
3. 对excel进行操作：
    获取excel的总行数、总列数、读取excel每一行的数据、读取excel每一列的数据、获取某个单元格的值
'''
#打开excel，打开的excel必须存在，返回book对象
book = xlrd.open_workbook('students.xlsx')
#通过索引获取sheet对象
sheet = book.sheet_by_index(0)
#有多个sheet页时可以通过sheet的名称来获取sheet对象
sheet1 = book.sheet_by_name('Sheet1')

#获取excel的总行数,
rows = sheet.nrows
#获取excel的总列数
cols = sheet.ncols
#获取excel第2行的数据，返回结果为list：[2.0, 'b', 'women']
row_value = sheet.row_values(2)
#获取excel第1列的数据，返回结果为list：['name', 'a', 'b', 'c', 'd', 'e', 'f', 'g', '小白', '小黑']
col_values = sheet.col_values(1)
#获取单元格第8行第1列的数据，返回结果为text： text:'小白'
cell_value = sheet.cell(8, 1)
#将text类型的结果转换为str类型：小白
cell_str = sheet.cell(8, 1).value
注意：获取每行、每列、某个单元格的值时，注意行、列的值要存在，否则会报错：list index out of range

excel - 读取excel小案例：

import xlrd
'''
读取excel的数据，读取数据的列固定，循环读取每行数据，读取后的数据格式如下：
[
{'name':xxx,'sex':xxx,'id':1},
{'name':xxx,'sex':xxx,'id':1},
.......
]
'''
def readExcel():
    try:
        #若输入的excel不存在，则打开excel报错
        book = xlrd.open_workbook('students.xlsx')
    except Exception as e:
        print('error msg:', e)
    else:
        sheet = book.sheet_by_index(0)
        #获取excel的总行数
        rows = sheet.nrows
        stu_list = []
        #循环读取每行数据，第0行是表头信息，所以从第1行读取数据
        for row in range(1, rows):
            stu = {}
            #获取第row行的第0列所有数据
            id = sheet.cell(row, 0).value
            name = sheet.cell(row, 1).value
            sex = sheet.cell(row, 2).value
            #将id、name、sex添加到字典，若元素不存在则新增，否则是更新操作
            stu['id'] = id
            stu['name'] = name
            stu['sex'] = sex
            stu_list.append(stu)
        print(stu_list)

if __name__ == '__main__':
    readExcel()

excel数据格式如下：

 excel - 写操作知识点：

import xlwt
'''
写 excel的操作步骤如下：
1. 打开excel，打开不存在的excel，若打开已存在的excel，进行写操作，写入的数据会覆盖以前的数据
2. 获取sheet对象并指定sheet的名称
3. 对excel进行操作：
    写入excel、保存excel
'''
#打开excel创建book对象
book = xlwt.Workbook()
#创建sheet指定sheet名称
sheet = book.add_sheet('stu2')
#写入excel数据，第n行第n列写入某个值，写入的数据类型为str
sheet.write(0, 0, '编号')
sheet.write(0, 1, '姓名')
sheet.write(0, 2, '年龄')
#保存excel,保存的后缀必须是xls
book.save('studet.xls')

excel写入 新的excel后，数据格式如下：

excel操作 已存在的excel，进行写操作后的 excel格式如下：

 ----> 

excel - 写excel小案例：

import xlwt
'''
将list数据：
[{'name': '小白', 'id': 1.0, 'sex': '男'},
    {'name': '小花', 'id': 2.0, 'sex': '女'},
    {'name': '小黑', 'id': 3.0, 'sex': '男'},
     {'name': '小茹', 'id': 4.0, 'sex': '女'},
      {'name': '小小', 'id': 5.0, 'sex': '男'}]
写入excel，title信息为：编号、姓名、性别
'''
def writeExcel():
    book = xlwt.Workbook()
    sheet = book.add_sheet('stu')
    titles = ['编号', '姓名', '性别']
    #循环读取titles的长度，col的值为：0,1,2,并将title值写入excel
    for title_col in range(len(titles)):
        #title 写入excel的第0行的第col列，写入titles[col]值
        sheet.write(0, title_col, titles[title_col])
    students_list = [{'name': '小白', 'id': 1.0, 'sex': '男'},{'name': '小花', 'id': 2.0, 'sex': '女'},{'name': '小黑', 'id': 3.0, 'sex': '男'},{'name': '小茹', 'id': 4.0, 'sex': '女'},{'name': '小小', 'id': 5.0, 'sex': '男'}]
    for stu_row in range(len(students_list)):
        #循环读取student_list的长度，从0开始，写入excel时从第1行开始写入数据
        #写入excel的数据是从list里进行取值，获取list的每个元素，返回字典，然后通过字典的key获取value
        sheet.write(stu_row+1, 0, students_list[stu_row]['id'])
        sheet.write(stu_row+1, 1, students_list[stu_row]['name'])
        sheet.write(stu_row+1, 2, students_list[stu_row]['sex'])
    book.save('student.xls')
if __name__ == '__main__':
    writeExcel()

excel数据格式如下：

 excel- 更新操作知识点：

import xlrd
from xlutils.copy import copy
'''
更新excel操作：
1. 打开excel，更新的excel必须存在
2. 复制一个新的excel，使用xlutils模块中的copy方法
3. 更新excel内的数据
4. 保存更新后的excel数据，以前的excel数据不会更改
'''
from xlutils.copy import copy
#打开excel
book = xlrd.open_workbook('student.xlsx')
#复制一个新的excel
new_book = copy(book)
#查看某个对象下的所有方法
#print(dir(new_book))
#获取新excel的sheet对象
sheet = new_book.get_sheet(0)
#新增一列数据
sheet.write(0, 3, '更新')
#更新第4行第1列的值，将其修改为'郭静'，修改的数据类型为str
sheet.write(4, 1, '郭静')
#保存更改后的excel，以前的excel数据不更改
new_book.save('student.xls')

 以上为excel简单操作~~~~

案例：

写一个函数，传入一个表名，然后把这个表里面所有的数据导出excel里面

import pymysql
import xlwt
import hashlib

dic = {
    "host": '192.1xx.xx.x',
    'user': 'mysql_xx',
    'password': 'xxx@123',
    'port': 3306,
    'db': 'user_xx'
}

#mysql 操作
def op_mysql( sql, res_many=False):
    con = pymysql.connect(host=dic['host'], user=dic['user'], password=dic['password'], db=dic['db'], port=dic['port'], charset='utf8', autocommit=True)
    cur = con.cursor(pymysql.cursors.DictCursor)
    cur.execute(sql)
    if res_many:
        res = cur.fetchall()
    else:
        res = cur.fetchone()
    cur.close()
    con.close()
    return res

# 查询sql，将结果写入excel
def op_table(table_name):
    sql = 'select * from {0};'.format(table_name)
    res_data = op_mysql(sql, res_many=True)  #返回结果是list [{},{}]

    book = xlwt.Workbook()
    sheet = book.add_sheet('标签')

    for row, row_data in enumerate(res_data):
        for col, col_key in enumerate(row_data):   # 获取下标、字典key
            if row == 0:
                sheet.write(0, col, col_key)
            else:
                sheet.write(row, col, row_data[col_key])

    book.save('label_data.xls')

op_table('xxx_label')

 excel 插件效果：