提交作业时，如何在Python文件中规范分割多个算法？

你的思路已经挺实用的了，但确实有更贴合Python最佳实践、扩展性更强的实现方式，这里给你几个常见的方案参考：

方案1：函数封装 + 字典映射（最适合小型脚本） #

把每个算法的逻辑封装成独立函数，然后用字典建立选择项和函数的映射，这样比一堆if-elif要简洁，后续加新算法也更方便：

# 封装算法1的逻辑
def run_algorithm_1():
    """执行算法1的逻辑"""
    print("Running Algorithm 1")
    # 这里写算法1的具体代码

# 封装算法2的逻辑
def run_algorithm_2():
    """执行算法2的逻辑"""
    print("Running Algorithm 2")
    # 这里写算法2的具体代码

# 建立选择与算法函数的映射
ALGORITHM_MAP = {
    "0": run_algorithm_1,
    "1": run_algorithm_2
}

if __name__ == "__main__":
    choice = input("0 for Algorithm 1 / 1 for Algorithm 2: ")
    # 校验输入有效性
    if choice in ALGORITHM_MAP:
        ALGORITHM_MAP[choice]()  # 调用对应的算法函数
    else:
        print("Invalid input! Please enter 0 or 1.")

这个方案的优势：

• 代码结构清晰，算法逻辑和选择逻辑分离
• 新增算法只需要添加函数和更新字典，不需要修改主逻辑的条件判断
• 易于维护和扩展

方案2：模块化拆分（适合大型项目） #

如果每个算法的逻辑比较复杂，建议把每个算法放到单独的模块文件中，主文件负责导入和调度，这样代码的可维护性和可读性会大幅提升：

比如项目结构：

your_project/
├── main.py
├── algorithm1.py
└── algorithm2.py

main.py：

import algorithm1
import algorithm2

ALGORITHM_MAP = {
    "0": algorithm1.run,
    "1": algorithm2.run
}

if __name__ == "__main__":
    choice = input("0 for Algorithm 1 / 1 for Algorithm 2: ")
    if choice in ALGORITHM_MAP:
        ALGORITHM_MAP[choice]()
    else:
        print("Invalid input! Please enter 0 or 1.")

algorithm1.py：

def run():
    """算法1的入口函数"""
    print("Running Algorithm 1")
    # 算法1的完整逻辑

algorithm2.py：

def run():
    """算法2的入口函数"""
    print("Running Algorithm 2")
    # 算法2的完整逻辑

这个方案的优势：

• 每个算法独立成模块，便于单独测试和调试
• 代码结构更清晰，适合多人协作或者算法逻辑复杂的场景

方案3：类封装（适合需要状态/参数配置的算法） #

如果你的算法需要初始化参数、维护运行状态，用类来封装会更合适：

class Algorithm1:
    def __init__(self, config=None):
        # 初始化算法1的配置参数
        self.config = config or {"threshold": 0.5}
    
    def run(self):
        """执行算法1"""
        print(f"Running Algorithm 1 with config: {self.config}")
        # 算法1的具体逻辑

class Algorithm2:
    def __init__(self, config=None):
        self.config = config or {"max_iter": 100}
    
    def run(self):
        """执行算法2"""
        print(f"Running Algorithm 2 with config: {self.config}")
        # 算法2的具体逻辑

ALGORITHM_MAP = {
    "0": Algorithm1,
    "1": Algorithm2
}

if __name__ == "__main__":
    choice = input("0 for Algorithm 1 / 1 for Algorithm 2: ")
    if choice in ALGORITHM_MAP:
        # 可以传入自定义配置，比如Algorithm1({"threshold": 0.7})
        alg_instance = ALGORITHM_MAP[choice]()
        alg_instance.run()
    else:
        print("Invalid input! Please enter 0 or 1.")

这个方案的优势：

• 可以很好地管理算法的状态和配置参数
• 支持更复杂的算法逻辑，比如需要多次运行、保存中间结果的场景

总结 #

• 如果是小型脚本，优先选方案1，简单高效
• 如果算法逻辑复杂或项目规模大，选方案2，模块化更易维护
• 如果算法需要状态管理或参数配置，选方案3，类封装更灵活

内容的提问来源于stack exchange，提问作者Wizard