python 执行时命令行传入参数

托马斯-木2024-06-04 17:01

例如要执行test 文件，想要将aaa 和 123 传入 test 作为一个变量传入执行

bash 复制代码

python test.py  aaa 123

方法 1 使用 argparse

python 复制代码

import argparse
if __name__ == "__main__":
  parser = argparse.ArgumentParser()
  parser.add_argument('--aa', type=str, default = None)
  parser.add_argument('--bb', type=int, default=32)
  args = parser.parse_args()
  if(args.bb==3):
    print(f'hello world, args.bb={args.bb}')
  else:
    print(f'sorry, args.bb is not 3, but ={args.bb}')

python 复制代码

import argparse
if __name__ == "__main__":
  parser = argparse.ArgumentParser()
  parser.add_argument('--aa', type=str, default = None)
  parser.add_argument('--bb', type=int, default=32)
  args = parser.parse_args()
  print(args.aa)
  print(args.bb)

方法 2 使用 sys.argv

python 复制代码

import sys
gpus = sys.argv[1]
batch_size = sys.argv[2]
print(gpus)
print(batch_size)

注意，可以使用以下代码获取输入命令个数

python 复制代码

len(sys.argv)

注意 sys.argv 为字符串，想要转换成int 性数据可以使用

python 复制代码

num=int(batch_size )

python 执行时 命令行传入参数

方法 1 使用 argparse

方法 2 使用 sys.argv

注意，可以使用以下代码 获取输入命令个数

注意 sys.argv 为字符串，想要转换成int 性数据可以使用

python 执行时命令行传入参数

注意，可以使用以下代码获取输入命令个数