如何使用 python3 argparse 将字符串值正确传递给用户定义的函数？

Question

我编写了一个短程序，使用正割法计算定义为可调用函数的各公司现金流量的内部利率 return。每个功能都以公司的股票行情或史诗命名。该程序是运行具有以下命令行参数：，并使用 argparse 解析命令行参数。示例：python3 SecantMethod.py grg 0.2 0.4 6。当程序运行s 产生“str object is not callable”错误时，到目前为止我一直无法解决。问题似乎与 argparse 传递所需现金流史诗的方式有关。删除所有 argparse 代码并插入一个简单的 print(SecantMethod(roo 0,1 0.2 6)) 语句导致程序 returning 正确的速率和迭代值。

import argparse


def SecantMethod(f, x0, x1, d):

    c = 0
    e = 0.5*pow(10, -d)

    while abs(x1 - x0) > e:
        if f(x1) - f(x0) == 0:
            return 'Division by zero'

        x = x1 - f(x1)*(x1 - x0)/(f(x1) - f(x0))
        x0, x1 = x1, x
        c += 1

    return x, c

def grg(x):

    return -1000\
       +100.00/(1 + x)\
       +210.00/(1 + x)**2\
       +331.00/(1 + x)**3\
       +446.10/(1 + x)**4\
      +1610.51/(1 + x)**5
    # root = 27.97% (x = 0.27967730)

def roo(x):

    return -4000\
       +1200/(1 + x)\
       +1410/(1 + x)**2\
       +1875/(1 + x)**3\
       +1050/(1 + x)**4
    # root = 14.30% (x = 0.14299344)


def main():

    parser = argparse.ArgumentParser(description= 'Solves f(x) = 0, using the secant method.')

    parser.add_argument('f', type=str, help='function name')
    parser.add_argument('x0', type=float, help='first interval endpoint')
    parser.add_argument('x1', type=float, help='second interval endpoint')
    parser.add_argument('d', type=int, help='number of decimal places')

    args = parser.parse_args()

    if args.x0 == args.x1:
        print('\nInvalid input: x0 and x1 must be distinct values.')
        return
    if args.d < 0:
        print('\nInvalid input: d must be a positive integer.')
        return

    # print('\nRoot & number of iterations: ', SecantMethod(roo, 0.1, 0.2, 8))
    print('\nRoot & number of iterations: ', SecantMethod(args.f, args.x0, args.x1, args.d))

if __name__ == "__main__":
    main()

Answer 1

字符串'grg'与函数grg不同。

处理此问题的一种方法是提供映射：

valid_functions = {
    'grg': grg,
    'roo': roo,
}
# ...
SecantMethod(valid_functions[args.f], ...)

如何使用 python3 argparse 将字符串值正确传递给用户定义的函数？

How to correctly pass string value to user defined function using python3 argparse?

python

argparse