Python3中的sys.argv

 

首先需要import sys，sys是python3的一个标准库 / 模块；

「argv」是「argument variable」参数变量的简写形式；是一个List列表。

argv[0] 一般是被调用的脚本文件名或全路径，和操作系统有关。 sys.argv[0]表示代码本身文件路径，

所以参数从1开始：argv[1]及之后是传入的数据。

下面这个例子中，我们先编写一个run_cnn.py文件，并定义好train 、test函数，然后再编写main函数，这样的话，我们就可以在terminal中分别输入 python run_cnn.py [train / test]来选择是训练还是测试模型

if __name__ == '__main__':

    if len(sys.argv) != 2 or sys.argv[1] not in ['train', 'test']:
        raise ValueError("""usage: python run_cnn.py [train / test]""")

    print('Configuring CNN model...')
    '''  模型配置'''

    if sys.argv[1] == 'train':
        train()
    else:
        test()