版权声明:转载请注明出处 https://blog.csdn.net/github_37412255/article/details/86412830
import openpyxl
from openpyxl import load_workbook
f = open(file='data/10886数据集/最原始数据/电子病历对应病历号from10886.txt', encoding='utf-8')
list_f = f.readlines()
ll = []
for item in list_f:
item = item.replace('\n', '')
ll.append(item)
wb = load_workbook("data/10886数据集/最原始数据/首次病程记录_抽取_原始数据_13665.xlsx")
sheet1 = wb.get_sheet_by_name("Sheet1")
# 读excel
column_A = sheet1["A"]
column_B = sheet1["B"]
column_C = sheet1["C"]
column_D = sheet1["D"]
column_E = sheet1["E"]
column_F = sheet1["F"]
column_G = sheet1["G"]
column_H = sheet1["H"]
filepath1 = 'data/10886数据集/最原始数据/从13665中抽取出的10886条数据.xlsx'
filepath2 = 'data/10886数据集/最原始数据/从13665中抽取出的除10886条数据以外的数据.xlsx'
wb1 = openpyxl.Workbook()
ws1 = wb1.create_sheet('Sheet1')
wb2 = openpyxl.Workbook()
ws2 = wb2.create_sheet('Sheet1')
index_ws1 = 1
index_ws2 = 1
# 写excel
for i in range(0, 13666):
print("i :" + str(i) + "...")
if i == 0:
ws1.cell(row=index_ws1, column=1, value=sheet1["A"][i].value).value
ws1.cell(row=index_ws1, column=2, value=sheet1["B"][i].value).value
ws1.cell(row=index_ws1, column=3, value=sheet1["C"][i].value).value
ws1.cell(row=index_ws1, column=4, value=sheet1["D"][i].value).value
ws1.cell(row=index_ws1, column=5, value=sheet1["E"][i].value).value
ws1.cell(row=index_ws1, column=6, value=sheet1["F"][i].value).value
ws1.cell(row=index_ws1, column=7, value=sheet1["G"][i].value).value
ws1.cell(row=index_ws1, column=8, value=sheet1["H"][i].value).value
ws2.cell(row=index_ws2, column=1, value=sheet1["A"][i].value).value
ws2.cell(row=index_ws2, column=2, value=sheet1["B"][i].value).value
ws2.cell(row=index_ws2, column=3, value=sheet1["C"][i].value).value
ws2.cell(row=index_ws2, column=4, value=sheet1["D"][i].value).value
ws2.cell(row=index_ws2, column=5, value=sheet1["E"][i].value).value
ws2.cell(row=index_ws2, column=6, value=sheet1["F"][i].value).value
ws2.cell(row=index_ws2, column=7, value=sheet1["G"][i].value).value
ws2.cell(row=index_ws2, column=8, value=sheet1["H"][i].value).value
index_ws1 += 1
index_ws2 += 1
else:
if str(sheet1["A"][i].value) in ll:
ws1.cell(row=index_ws1, column=1, value=sheet1["A"][i].value).value
ws1.cell(row=index_ws1, column=2, value=sheet1["B"][i].value).value
ws1.cell(row=index_ws1, column=3, value=sheet1["C"][i].value).value
ws1.cell(row=index_ws1, column=4, value=sheet1["D"][i].value).value
ws1.cell(row=index_ws1, column=5, value=sheet1["E"][i].value).value
ws1.cell(row=index_ws1, column=6, value=sheet1["F"][i].value).value
ws1.cell(row=index_ws1, column=7, value=sheet1["G"][i].value).value
ws1.cell(row=index_ws1, column=8, value=sheet1["H"][i].value).value
index_ws1 += 1
elif str(sheet1["A"][i].value) not in ll:
ws2.cell(row=index_ws2, column=1, value=sheet1["A"][i].value).value
ws2.cell(row=index_ws2, column=2, value=sheet1["B"][i].value).value
ws2.cell(row=index_ws2, column=3, value=sheet1["C"][i].value).value
ws2.cell(row=index_ws2, column=4, value=sheet1["D"][i].value).value
ws2.cell(row=index_ws2, column=5, value=sheet1["E"][i].value).value
ws2.cell(row=index_ws2, column=6, value=sheet1["F"][i].value).value
ws2.cell(row=index_ws2, column=7, value=sheet1["G"][i].value).value
ws2.cell(row=index_ws2, column=8, value=sheet1["H"][i].value).value
index_ws2 += 1
wb1.save(filepath1)
wb2.save(filepath2)