例如要执行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 )