Question

如果要从控制台/ shell调用脚本时提供非整数或非浮点值作为脚本的参数，是否有任何简便的方法可以返回各自的默认值？

import argparse

def check(v):
    try:
        return float(v)
    except:
        #return default value for respective index in default list

parser = argparse.ArgumentParser()

parser.add_argument('-data',dest='data', type = check, nargs=5,
                    default=[-1,1,-7.9,512,0])
data = args.data
print(data)

例如，如果提供了以下参数：

file.py -data 1 6.03 1 p 1

我只希望替换p，所以data将打印[1, 6.03, 1, 512, 1]。谢谢。

Answer 1

我认为最好的方法是将参数5分为5个不同的参数--data1 --data2（或者甚至为每个数据点提供更好的描述性名称），因为现在它是有位置的，因此很难衡量哪个参数没有填写。您也可能像在示例中所做的那样，传递了一个虚拟值（如“ p”），然后在解析args时可以将其替换为预设的默认值。

my_data = [default[i] if x == 'p' else x for i, x in enumerate(data)]

Answer 2

您可以在ValueError上引发type，然后使用sys.argv获取所有参数，使用非浮点/非整数值获取arg的索引，并替换为该索引的默认值。 defaults是字典，列表中包含默认值。

import argparse
import sys

def check(v):
    try:
        value = float(v)
        return value
    except ValueError:     #raise ValueError if the arg is not int           
        defaults = vars(parser.parse_args([]))   #get all the default values as a dict
        args = sys.argv[2:]         #exclude args at 0 and 1 index
        v_index = args.index(v)     #grab index at v              
        v = defaults['data'][v_index]  #assign the default value at v's index to 'v'
        return v

parser = argparse.ArgumentParser()
parser.add_argument('-data', dest='data',type = check, nargs=5, default=[-1,1,-7.9,512,0]) 
args = parser.parse_args() 
print(args.data)

现在使用您提供的参数进行测试。我的文件名为test.py：

python test.py -data 1 6.03 1 p 1
#Output:
[1.0, 6.03, 1.0, 512, 1.0]

让我们添加另一个非int / non-float arg：

python test.py -data 1 6.03 x p 1
#Output:
[1.0, 6.03, -7.9, 512, 1.0]

编辑：以下是一个通用的解决方案，与提供的标志（'-data'，'-name'）无关：

def check(v):
    #<--same-->
    except ValueError:             
        defaults = vars(parser.parse_args([]))   
        args = sys.argv[2:]                 
        arg_spec = str(sys.argv[1][1:]) #get the flag specified, '-name' etc.        
        v_index = args.index(v)     #grab index at v              
        v = defaults[arg_spec][v_index]  #assign the default value at v's index to 'v'
        return v

#add another flag to the arguments, 
#identify the flag passed with sys.argv, 
#convert the args passed into a dict,
#select the flag key to get its value(the args for the flag) from the dict

parser = argparse.ArgumentParser()
parser.add_argument('-data', dest='data',type = check, nargs=5, default=[-1,1,-7.9,512,0]) 
parser.add_argument('-name', dest='name',type = check, nargs=5, default=[5,20,606,81,15])
args = parser.parse_args() 
arg_spec = sys.argv[1][1:]
args_dict = vars(args)
print("The flag you provided is: '{}'. The args are: {}".format(sys.argv[1], args_dict[arg_spec]))

一些测试：

python test.py -name 1 6.03 1 x 1
#Output:
The flag you provided is: '-name'. The args are: [1.0, 6.03, 1.0, 81, 1.0]

python test.py -data 1 6.03 1 x 1
#Output:
The flag you provided is: '-data'. The args are: [1.0, 6.03, 1.0, 512, 1.0]

在某些情况下，使python argparse仅使用部分默认值

2 个答案: