时间戳

首页 > 解决方案 > 更有效的选择函数的方法

python - 更有效的选择函数的方法

问题描述

我正在尝试制作一个 Python 脚本，让我可以选择要执行的功能（例如，清理下载文件夹或下载视频）。所有这些案例都有一个单独的脚本，我将其导入到主 .py 文件中。我目前设置它的方式，我得到输入，然后如果它是 1 则执行第一个函数，如果它是 2 则执行第二个函数，依此类推。这是我目前的代码：

    if action == 1:
        instagram_downloader()
        print("Done")
    elif action == 2:
        clean_downloads()
        print("Done")

虽然这对于少量功能来说还不错，但一旦我进入两位数，它就会变得混乱和低效。有没有更好更有效的方法来做到这一点？

标签： python

解决方案

是的，使用字典。这相当于 Python 中的其他语言中的 case/switch 语句。

functions = {1: instagram_downloader, 2: clean_downloads}
function = functions[action]
function()

推荐阅读