python 执行时 命令行传入参数

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

python test.py  aaa 123

方法 1 使用 argparse

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}')

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

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

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

len(sys.argv)   

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

num=int(batch_size )