定义命令行参数为整数类型的方法(使用DEFINE_integer())
发布时间:2024-01-05 10:16:36
DEFINE_integer()是Google的命令行解析库gflags提供的一个函数,用于定义整数类型的命令行参数。
使用DEFINE_integer()函数定义整数类型的命令行参数需要提供两个参数:参数名和默认值。
定义命令行参数的示例代码如下:
from absl import app
from absl import flags
FLAGS = flags.FLAGS
def main(argv):
# 定义整数类型的命令行参数
flags.DEFINE_integer('num_iterations', 10, 'Number of iterations')
# 解析命令行参数
flags.parse_args(argv)
# 获取命令行参数的值
num_iterations = FLAGS.num_iterations
# 打印命令行参数的值
print('Num iterations:', num_iterations)
if __name__ == '__main__':
app.run(main)
在上面的示例中,我们通过调用flags.DEFINE_integer()函数,定义了一个整数类型的命令行参数num_iterations,默认值为10。
然后,通过调用flags.parse_args()函数,解析命令行参数。
最后,在main()函数中,我们获取并打印了命令行参数num_iterations的值。
要运行以上代码,并指定命令行参数的值,可以在命令行中运行以下命令:
python my_script.py --num_iterations=20
运行结果将输出:
Num iterations: 20
如果没有指定命令行参数num_iterations的值,则使用默认值10。
通过使用DEFINE_integer()函数,我们可以方便地定义整数类型的命令行参数,并在代码中使用这些命令行参数的值。
