如何根据python中给定的命令行输入来运行某些函数

import argparse

# 定义一些示例函数
def function_a():
    print("Running Function A")

def function_b():
    print("Running Function B")

def function_c():
    print("Running Function C")

# 创建 ArgumentParser 对象
parser = argparse.ArgumentParser(description="Run different functions based on command line input.")

# 添加位置参数
parser.add_argument('function', choices=['a', 'b', 'c'], help="Function to run (a, b, or c)")

# 解析命令行参数
args = parser.parse_args()

# 根据解析到的参数调用相应的函数
if args.function == 'a':
    function_a()
elif args.function == 'b':
    function_b()
elif args.function == 'c':
    function_c()

python script.py a
# 输出: Running Function A

python script.py b
# 输出: Running Function B

python script.py c
# 输出: Running Function C