python中excel操作

python中excel操作

# coding=utf-8
"""
excel操作使用openpyxl
作用：excel存测试数据
一.安装第三方库
pip install openpyxl==3.0.7
二.excel 组成
1.excel文件==表格对象   一个表格对象有多个表单对象   小区大门
2.表单==表单对象        一个表单对象有多个单元格对象  单元门
3.单元格==单元对象       单元格对象是excel的最小组成部分   你家的门
三. 操作步骤
from openpyxl import load_workbook

# 第一步创建表格对象
wb = load_workbook(filename='cases.xlsx')
# 第二步获取表单对象
sheet_obj = wb["Sheet1"]
# 第三步获取单元格对象
cell_obj = sheet_obj["B3"]
cell_obj2 = sheet_obj.cell(4, 2)
print(cell_obj)
# 第四步获取对应单元格值
print(cell_obj.value)
print(cell_obj2.value)
# 第五步关闭excel连接
wb.close()

四.行操作
1.result =sheet_obj.iter_rows()
#min_row=None, max_row=None, min_col=None, max_col=None, values_only=False
min_row=None : 最小的行索引值(索引从1开始，必须是int类型，默认是1)
max_row=None : 最大的行索引值(索引从1开始，必须是int类型，默认是最大的行)
min_clo=None : 最小的列索引值(索引从1开始，必须是int类型，默认是1)
max_col=None : 最大的列索引值(索引从1开始，必须是int类型，默认是最大的列)
values_only=False  ：True获取到的是value值，False拿到的是对象需要通过value属性获取对应的值
注意点:切片是闭区间，包含起始和结束索引对应的行的数据
2.获取最大行
print(sheet_obj.max_row)
3.获取最大的列
print(sheet_obj.max_column)

五.列操作
result = sheet_obj.iter_cols(min_col=1, max_col=6, values_only=True)
wb = load_workbook(filename='cases.xlsx', read_only=False)
read_only=False: 可以直接获取列
read_only=True: 如果是True，列切片不可以直接读取值



"""
from openpyxl import load_workbook

wb = load_workbook(filename='cases.xlsx', read_only=False)
# read_only =False: 如果是True,列切片不可以直接读取值
sheet_obj = wb['Sheet1']
# 行切片
# 获取所有的表单名称
# print(wb.sheetnames)
# 根据名称获取对应的表单对象
# for i in wb.sheetnames:
#     print(wb[i])
# False拿到的是对象需要通过value属性获取对应的值
# result = sheet_obj.iter_rows(min_row=1, max_row=3, min_col=1, max_col=6, values_only=False)
# for i in result:
#     for j in i:
#         print(j.value)
# True获取到的是value值
# result = sheet_obj.iter_rows(min_row=1, max_row=4, min_col=1, max_col=6, values_only=True)
# print(list(result))
# 列切片
# result = sheet_obj.iter_cols(min_col=1, max_col=6, values_only=False)
# for i in result:
#     for j in i:
#         print(j.value)
# read_only=False(必须是False):可以直接获取列的值
result = sheet_obj.iter_cols(min_col=1, max_col=6, values_only=True)
print(list(result))
wb.close()

2.线性脚本(重点）

# codingg=utf-8
from openpyxl import load_workbook


class Excel:
    def __init__(self, name, sheet_name):
        self.wb = load_workbook(name, read_only=False)
        self.sheet_obj = self.wb[sheet_name]

    def get_data(self):
        data_list = []
        result = list(self.sheet_obj.iter_rows(values_only=True))
        result_title = result[0]
        result_value = result[1:]
        for i in result_value:
            result_dict = list(zip(result_title, i))
            data_list.append(result_dict)
        self.close_file()
        return data_list

    def close_file(self):
        self.wb.close()


if __name__ == '__main__':
    from pprint import pprint

    excel = Excel('cases.xlsx', 'Sheet1')
    result = excel.get_data()
    pprint(result)

Excel 写入

from openpyxl import load_workbook

# 1.获取表格对象
wb = load_workbook(filename='cases.xlsx')
# 2.获取表单对象
sheet_obj = wb['Sheet1']
# 3.写入数据(覆盖)
# sheet_obj['A1'] = 'py52'
# sheet_obj.cell(row=1, column=2, value='test_value')
# 追加写入(不覆盖数据)
# 接受可迭代对象：list,tuple,dict,生成器
test_list = [1, 2, 3, 4, 5]
test_dict = {'A': 'test1', 'B': 'test2'}
sheet_obj.append(test_dict)
wb.save('cases.xlsx')
print(sheet_obj['B1'].value)
wb.close()

如果代码有错误，希望大家提出来