import openpyxl
import os
defwrite_xlsx(filename, sheet_index, sheet_name, data):"""
Judge whether the file exists,
if it exists, open the original file,
otherwise create a new file
:param filename: Target file path
:param sheet_index: create a new sheet's index
:param sheet_name: create a new sheet's name
:param data: add data
"""if os.path.exists(filename):"the file has already exists"
wb = openpyxl.load_workbook(filename)else:"not exists"
wb = openpyxl.Workbook()
sheet = wb.create_sheet(title=sheet_name, index=sheet_index)for item in data:
sheet.append(item)
wb.save(filename)if __name__ =='__main__':
filename ="test.xlsx"
sheet_index =0
sheet_name ="test_sheet1"
data =[[1,"小明"],[2,"小红"]]
write_xlsx(filename, sheet_index, sheet_name, data)