用 Python 把 EXCEL 题库 转Word文档

from docx import Document
import openpyxl
import random

def generate_exam():
    # 单选题随机抽取
    def danxuan(num_questions):
        workbook = openpyxl.load_workbook('danxuan.xlsx')
        sheet = workbook.active
        questions = []
        for row in sheet.iter_rows(min_row=2, max_row=sheet.max_row, values_only=True):
            question = {
                '类型': row[0],
                '序号': row[1],
                '题干': row[2],
                'A': row[3],
                'B': row[4],
                'C': row[5],
                'D': row[6],
                '正确答案': row[7]
            }
            questions.append(question)

        selected_questions = random.sample(questions, num_questions)
        return selected_questions

    # 判断题随机抽取
    def panduan(num_questions):
        workbook = openpyxl.load_workbook('panduan.xlsx')
        sheet = workbook.active
        questions = []
        for row in sheet.iter_rows(min_row=2, max_row=sheet.max_row, values_only=True):
            questions.append(row)
        return random.sample(questions, num_questions)

    # 多选题随机抽取
    def duoxuan(num_questions):
        workbook = openpyxl.load_workbook('duoxuan.xlsx')
        sheet = workbook.active
        questions = []
        for row in sheet.iter_rows(min_row=2, max_row=sheet.max_row, values_only=True):
            questions.append(row)
        return random.sample(questions, num_questions)

    # 定义题目数量
    num_single_choice = 10
    num_true_false = 15
    num_multiple_choice = 15

    # 创建Word文档
    doc = Document()

    # 抽取题目
    selected_single_choice = danxuan(num_single_choice)
    selected_true_false = panduan(num_true_false)
    selected_multiple_choice = duoxuan(num_multiple_choice)

    # 生成试卷
    doc.add_paragraph("一、单选题")
    for i, question in enumerate(selected_single_choice, start=1):
        question_content = question['题干']
        option_A = question['A']
        option_B = question['B']
        option_C = question['C']
        option_D = question['D']
        doc.add_paragraph(f"{i}.  {question_content}")
        doc.add_paragraph(f"A. {option_A}      B. {option_B}      C. {option_C}      D. {option_D}")

    doc.add_paragraph("二、判断题")
    for i, question in enumerate(selected_true_false, start=1):
        question_content = question[1]
        doc.add_paragraph(f" {i}. {question_content}")

    doc.add_paragraph("三、多选题")
    for i, question in enumerate(selected_multiple_choice, start=1):
        question_content = question[1]
        option_A = question[2]
        option_B = question[3]
        option_C = question[4]
        option_D = question[5]
        doc.add_paragraph(f"{i}.  {question_content}")
        doc.add_paragraph(f"A. {option_A}      B. {option_B}      C. {option_C}      D. {option_D}")

    # 添加答案部分
    doc.add_page_break()
    doc.add_paragraph("单选题答案：")
    answers = ""
    for i, question in enumerate(selected_single_choice, start=1):
        correct_answer = question['正确答案']
        answers += f" {i}. {correct_answer}  "
    doc.add_paragraph(answers)

    doc.add_paragraph("判断题答案：")
    answers = ""
    for i, question in enumerate(selected_true_false, start=1):
        correct_answer = question[2]
        answers += f" {i}. {correct_answer}  "
    doc.add_paragraph(answers)

    doc.add_paragraph("多选题答案：")
    answer = ""
    for i, question in enumerate(selected_multiple_choice, start=1):
        correct_options = question[-1]
        answer += f" {i}. {correct_options}  "
    doc.add_paragraph(answer)

    # 保存生成的试卷文件
    doc.save("试卷generated_exam.docx")

    print("试卷已生成")

if __name__ == '__main__':
    generate_exam()