Python实现批量导入1000条xlsx数据

2023-02-16 12:02:40 数据 导入 批量

遇到的问题:

用户批量导入数据1000条,导入不成功的问题,提示查询不到商品资料。这个场景需要依靠批量的数据,每次测试的时候需要手动生成批量的数据,然后再导入操作,费时费劲。所以写了个脚本来实现。

在前面有说过用python实现导入接口请求,现在来说一下xlsx中1000条自动生成:

xlsx数据离不开openpyxl库,openpyxl库有个Workbook对象,其代表一个excel文档,Workbook中有一张或多张sheet,一个sheet就是一个Worksheet对象。

首先需要导包,from openpyxl import load_workbook,from openpyxl import Workbook。

xlsx文件的新增:

from openpyxl import load_workbook
import xlsxwriter
 
file_name = "max_1000.xlsx"
# 新建xlsx文件
workbook = xlsxwriter.Workbook(file_name)
worksheet = workbook.add_worksheet('Sheet1')
worksheet.write_row("A1",
                    ['公司', '编码', '渠道'])
#新增1000条数据
while m < 1000:
    m += 1
    # ["18"]、["21"]、["8"]分别代表第m行第1,2,3列的值,这里是写死的数据,也可以是参数化的数据
    worksheet.write_row(m, 0, ["18"])
    worksheet.write_row(m, 1, ["21"])
    worksheet.write_row(m, 2, ["8"])
# 关闭文件
workbook.close()
# 执行批量导入
data = load_yaml('./max_1000.yaml')['upload_file']
execute(data)

xlsx文件的读取:

通过Cell对象读取每一格内容

wb = load_workbook('max_1000.xlsx')
sheets = wb.worksheets
# 获取第一个sheet
sheet1 = sheets[0]
# 获取sheet1表第2行第1列的数据
cell_21 = sheet1.cell(2,1).value
# 获取sheet1表第3行第2列的数据
cell_32 = sheet1.cell(3,2).value

xlsx文件新增后,可用于批量导入的数据,xlsx文件的读取可用于数据的清理,比如已导入数据的清除。

也可用for循环,读取xlsx文件的所有数据。

xlsx文件读取所有数据:

wb = load_workbook("max_1000.xlsx")
sheets = wb.worksheets
Sheet1 = sheets[0]
# 循环读取数据,Sheet1.max_row代表表格的最大行数
for i in range(Sheet1.max_row - 1):
    # 获取Sheet1表中所有行第4列的数据
    data = Sheet1.cell(i + 2, 4).value

xlsx文件的编辑:

wb = load_workbook(file_path)

ws = wb[sheet]

# 修改.xlsx文件中的数据,cell[0]对应行数, cell[1]对应列数,cell[-1]对应需要修改后的值

ws.cell(cell[0], cell[1]).value = cell[-1]

wb.save(file_path)

wb.close()

到此这篇关于Python实现批量导入1000条xlsx数据的文章就介绍到这了,更多相关Python 批量导入xlsx数据内容请搜索以前的文章或继续浏览下面的相关文章希望大家以后多多支持!

相关文章