亲宝软件园·资讯

展开

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

qq_41845402 人气:0

遇到的问题:

用户批量导入数据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()

加载全部内容

相关教程
猜你喜欢
用户评论