在Excel中按条件筛选数据并存入新的表
实现代码:
# 1.获取满足条件的数据
from openpyxl import load_workbook
wb = load_workbook("每月物料表.xlsx")
data = {} # 储存所有工作表中满足条件的数据,以工作表名称为键
sheet_names = wb.sheetnames
for sheet_name in sheet_names:
ws = wb[sheet_name]
qty_list = []
# 获取G列的数据,并用enumrate给其对应的元素编号
for row in range(2, ws.max_row + 1):
qty = ws['G' + str(row)].value
qty_list.append(qty)
# print(qty_list)
qty_idx = list(enumerate(qty_list)) # 用于编号
# 判断数据是否大于1000,然后返回大于1000的数据所对应的行数
row_idx = [] # 用于储存数量大于1000所对应的的行号
for i in range(len(qty_idx)):
if qty_idx[i][1] > 1000:
row_idx.append(qty_idx[i][0] + 2)
# print(row_idx)
# 获取满足条件的数据
data_morethan1K = []
for i in row_idx:
data_morethan1K.append(ws['A' + str(i) + ":" + 'I' + str(i)])
# print(data_morethan1K)
data[sheet_name] = data_morethan1K
# 2.写入获取的数据
from openpyxl.styles import Border, Side, PatternFill, Font, GradientFill, Alignment
thin = Side(border_style="thin", color="000000") # 定义边框粗细及颜色
wb = load_workbook("模板.xlsx")
ws = wb.active
for month in data.keys():
ws_new = wb.copy_worksheet(ws) # 复制模板中的工作表
ws_new.title = month
print(ws_new.title)
# 将每个月的数据条数逐个取出并写入新的工作表
for i in range(len(data[month])): # 按数据行数计数,每行数据对应9列,所以每行需分别写入9个单元格
ws_new.cell(row=i + 2, column=1).value = data[month][i][0][0].value
ws_new.cell(row=i + 2, column=2).value = data[month][i][0][1].value
ws_new.cell(row=i + 2, column=3).value = data[month][i][0][2].value
ws_new.cell(row=i + 2, column=4).value = data[month][i][0][3].value.date()
ws_new.cell(row=i + 2, column=5).value = data[month][i][0][4].value
ws_new.cell(row=i + 2, column=6).value = data[month][i][0][5].value
ws_new.cell(row=i + 2, column=7).value = data[month][i][0][6].value
ws_new.cell(row=i + 2, column=8).value = data[month][i][0][7].value
ws_new.cell(row=i + 2, column=9).value = data[month][i][0][8].value
# 设置字号,对齐,缩小字体填充,加边框
# Font(bold=True)可加粗字体
for row_number in range(2, ws_new.max_row + 1):
for col_number in range(1, 10):
c = ws_new.cell(row=row_number, column=col_number)
c.font = Font(size=10)
c.border = Border(top=thin, left=thin, right=thin, bottom=thin)
c.alignment = Alignment(horizontal="left", vertical="center", shrink_to_fit=True)
wb.save("每月(大于1K).xlsx")