python脚本传参-编程知识

sys.argv
argparse

第一种：argparse

简单使用：

import argparse # 创建一个参数解析实例
parser = argparse.ArgumentParser(description='Parameters') # 添加参数解析
parser.add_argument('--training_epoch', type=int, default=3000)
parser.add_argument('--learning_rate', type=float, default=0.001)
parser.add_argument('--model_name', type=str, default='lstm')# 开始解析args = parser.parse_args(args=[]) # 输出参数
print({args})
print(args.training_epoch)
print(args.learning_rate)
print(args.model_name)

parser.add_argument('--go', help='input file path')

name or flags:
普通参数或flag参数选项参数的名称或标签，例如 epochs 或者 -e, --epochs。Flag参数不需要指定参数值，只需要带有参数名即可。
action: 命令行遇到flags参数时的动作。有两个常见的动作，store_true：设定flag参数为true；-store_false：设定flag参数为False。注意：如果直接运行程序，默认不读取该变量，要使用必须要进行传参，例如：python try.py --epochs
nargs: 应该读取的命令行参数个数，可以是具体的数字，或者是?号，当不指定值时对于 Positional argument 使用 default，对于 Optional argument 使用 const；或者是 * 号，表示 0 或多个参数；或者是 + 号表示 1 或多个参数。
default: 不指定参数时该参数的默认值。
type: 命令行参数应该被转换成的数据类型。
required: 是否为必选参数或可选参数。
help: 参数的帮助信息。
metavar：在 usage 说明中的参数名称，对于必选参数，默认就是参数名称，对于可选参数默认是全大写的参数名称。
dest：解析后的参数名称，默认情况下，对于可选参数选取最长的名称，中划线转换为下划线.
choices：参数可允许的值的一个容器。
const： action 和 nargs 所需要的常量值。
store_const：表示赋值为const；
append：将遇到的值存储成列表，也就是如果参数重复则会保存多个值;
append_const：将参数规范中定义的一个值保存到一个列表；
count：存储遇到的次数；此外，也可以继承 argparse.Action 自定义参数解析；

可以给KerasReadModel.py直接控制台传参数：

import argparse
parser = argparse.ArgumentParser()
## nargs='+'接受1个或多个参数，
## nargs='*'接受零个或多个
parser.add_argument('--list', nargs='+', help='<Required> Set flag', required=True)
args = parser.parse_args()if __name__ == '__main__':print(args.list)

运行结果如下:

import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--ll', action='append', help='<Required> Set flag', required=True)
args = parser.parse_args()if __name__ == '__main__':print(args.ll)

运行示例及结果:

亲测：讲文件KerasReadModel.py进行打包KerasReadModel.exe，然后控制台传入参数

import tensorflow as tf
from sklearn.preprocessing import StandardScaler
import argparse
import numpy as npdef readModel():parser = argparse.ArgumentParser()parser.add_argument('--test_data', type=float, nargs='+', required=True)args = parser.parse_args()test_array = np.array(args.test_data)  # 将传入的list转为array类型print('test_array:',test_array)ne_test_array=test_array.reshape(1,7).tolist()print('ne_test_array:', ne_test_array)# data_x = [[1, 80, 170, 50, 370, 79.3, 0]]sc = StandardScaler()x_data = sc.fit_transform(ne_test_array)new_model = tf.keras.models.load_model('./saved_model/blood_model', compile=False,custom_objects={"TrajGRUCell": 'TrajGRUCell', "TrajGRU": 'TrajGRU'})y_pred = new_model.predict(x_data)print(y_pred)if __name__ == "__main__":readModel()

注意reshape用法记录一下：

# numpy中reshape函数的三种常见相关用法
#
# reshape(1,-1)转化成1行：
#
# reshape(2,-1)转换成两行：
#
# reshape(-1,1)转换成1列：
#
# reshape(-1,2)转化成两列
# 如果一个 ndarray 有 个元素，那么它的 reshape(-1, 1, 2) 方法将返回一个 shape = (n, 1, 2) 的 ndarray.

打包文件和模型放到一起：

控制台执行命令：

E:\开发工具\pythonProject\flaskTest\dist>KerasReadModel.exe --test_data 0 80 170 50 370 79 0

成功！

第二种：sys.args

实际测试，传递列表

def readModel():test_data=sys.argv[1:]print('test_data:', test_data)test_array = np.array(test_data)  # 将传入的list转为array类型print('test_array:',test_array)ne_test_array=test_array.reshape(1,7).tolist()# data_x = [[1, 80, 170, 50, 370, 79.3, 0]]sc = StandardScaler()x_data = sc.fit_transform(ne_test_array)new_model = tf.keras.models.load_model('./saved_model/blood_model', compile=False,custom_objects={"TrajGRUCell": 'TrajGRUCell', "TrajGRU": 'TrajGRU'})y_pred = new_model.predict(x_data)print(y_pred)if __name__ == "__main__":readModel()

(venv) E:\开发工具\pythonProject\flaskTest>KerasReadModel.py 1 80 170 50 370 79.3 0

详细用法来自网上：

一、sys.argv[0]
sys.argv[0] 用来获取执行的脚本名称。

# test.py
import sysdef receive_args():file_name = sys.argv[0]print file_namereceive_args()执行结果：
#python test.py 
test.py

二、sys.argv[:]
sys.argv[:]用来获取参数列表，注意sys.argv[1:]这样可以去掉脚本名称

import sysdef receive_args():file_name = sys.argv[0]all_args = sys.argv[:]print file_nameprint all_argsreceive_args()#python test.py "2023-03-14" "liming" "shopping"
test.py
['test.py', '2023-03-14', 'liming', 'shopping']

三、sys.argv[n]
n 代表数字1,2,3… 表示第几个参数，用来获取指定参数

import sysdef receive_args():file_name = sys.argv[0]all_args = sys.argv[:]time = sys.argv[1]who = sys.argv[2]do_what = sys.argv[3]print file_nameprint all_argsprint timeprint whoprint do_whatreceive_args()#python test.py "2023-03-14" "liming" "shopping"
test.py
['test.py', '2023-03-14', 'liming', 'shopping']
2023-03-14
liming
shopping

四、利用列表其他属性
下面仅使用append()方法

import sysdef receive_args():file_name = sys.argv[0]all_args = sys.argv[:]time = sys.argv[1]who = sys.argv[2]do_what = sys.argv[3]print file_nameprint all_argsprint timeprint whoprint do_whatall_args.append("morning")print all_argsreceive_args()#python test.py "2023-03-14" "liming" "shopping"
test.py
['test.py', '2023-03-14', 'liming', 'shopping']
2023-03-14
liming
shopping
['test.py', '2023-03-14', 'liming', 'shopping', 'morning']