for index1, itme1 in enumerate(list_col): for index, itme in enumerate(list_row): if itme1 == '总计': count = count_conditions( teacher_arr, lambda row: row[9] == '学前' or row[9] == '小学', lambda row: row[8] == itme) if count != 0: sh_4149.cell(index1+6, index+4).value = count if itme1 == '女': count = count_conditions( teacher_arr, lambda row: row[2] == '女', lambda row: row[8] == itme) if count != 0: sh_4149.cell(index1+6, index+4).value = count if itme1 == '少数民族': count = count_conditions( teacher_arr, lambda row: row[11] != '汉族', lambda row: row[8] == itme) if count != 0: sh_4149.cell(index1+6, index+4).value = count if itme1 == '在编人员': count = count_conditions( teacher_arr, lambda row: row[12] == '是', lambda row: row[8] == itme) if count != 0: sh_4149.cell(index1+6, index+4).value = count代码优化
时间: 2023-06-09 11:07:31 浏览: 42
我理解您的问题是如何对上述代码进行优化。这段代码中有多个重复的条件判断语句,可以将其抽取出来,统一处理。同时,可以使用列表推导式简化代码。
以下是优化后的代码:
count_funcs = [
(lambda row: row[9] == '学前' or row[9] == '小学'), # 总计
(lambda row: row[2] == '女'), # 女
(lambda row: row[11] != '汉族'), # 少数民族
(lambda row: row[12] == '是') # 在编人员
]
for index1, item1 in enumerate(list_col):
for index, item in enumerate(list_row):
for count_func in count_funcs:
if item1 == '总计':
count = count_conditions(teacher_arr, count_func, lambda row: row[8] == item)
elif item1 == '女':
count = count_conditions(teacher_arr, lambda row: row[2] == '女' and count_func(row), lambda row: row[8] == item)
elif item1 == '少数民族':
count = count_conditions(teacher_arr, count_func, lambda row: row[8] == item)
elif item1 == '在编人员':
count = count_conditions(teacher_arr, count_func, lambda row: row[8] == item)
else:
count = 0
if count != 0:
sh_4149.cell(index1 + 6, index + 4).value = count
该代码将条件判断语句抽取出来,存放在一个列表中,然后在循环中依次调用。对于 '女' 这种特殊情况,将其与对应的条件判断函数同时传入 count_conditions 函数中。同时,使用加法替代了 index1 6 和 index 4。列表推导式被用来初始化 count_funcs 列表,使其更加简洁。