如何使用Argparse实现预测网络的命令行参数输入替代交互式输入？

没问题，我来帮你把脚本里的交互式输入改成通过命令行传参的方式，用argparse库就能轻松搞定，具体操作步骤如下：

步骤1：导入Argparse库 #

首先在你的脚本开头（所有导入语句的位置）加上这行代码：

import argparse

步骤2：创建参数解析器并定义参数 #

在脚本中原来的交互式输入代码之前，添加以下代码来定义命令行参数：

# 创建解析器对象，添加描述信息方便用户理解脚本用途
parser = argparse.ArgumentParser(description='预测网络：输入密度比和表面粗糙度以获取预测值')

# 添加两个位置参数，分别对应你的两个输入参数
# 第一个参数是密度比，类型为float，help是参数的说明文字
parser.add_argument('density', type=float, help='Density Ratio (%)')
# 第二个参数是表面粗糙度Ra，同样设置类型为float
parser.add_argument('ra', type=float, help='Surface Roughness- Ra (um)')

步骤3：解析命令行参数并替换原有输入代码 #

接下来解析用户传入的命令行参数，然后替换你第42、43行的交互式输入代码：

# 解析命令行参数
args = parser.parse_args()

# 替换原来的input语句，直接从解析后的参数中取值
Density = args.density
Ra = args.ra

步骤4：运行脚本的方式 #

现在你就可以在命令行中直接传入参数启动脚本了，比如：

python script.py 10 20

这里的10就是你要传入的Density Ratio，20是Surface Roughness- Ra的值。

如果用户想查看参数说明，只需要运行：

python script.py -h

就会显示你之前定义的参数帮助信息，非常友好。

这样修改后，脚本就不再需要运行过程中等待用户输入，而是直接从命令行获取参数，更适合自动化运行或者批量测试的场景。

内容的提问来源于stack exchange，提问作者Tony