Python | 接受脚本输入

在Python中，可以轻松地接受命令行中的脚本输入。接受脚本输入是一个非常常见的任务，它让我们的脚本变得更加灵活和通用。

从命令行接受脚本输入

接受脚本输入的方式就是通过使用sys.argv获取命令行参数。 例如，如果我们有以下脚本：

import sys

print("命令行参数为：", sys.argv)

我们可以通过以下方式将参数传递给脚本：

python script.py argument1 argument2

以上命令将在脚本中显示以下结果：

命令行参数为： ['script.py', 'argument1', 'argument2']

argparse

Python内置了一个名为argparse的模块，可以帮助我们更好地组织和解析命令行参数。argparse可以轻松地定义和解析命令行参数，还可以生成帮助和使用文档。

下面是一个示例脚本，演示如何使用argparse解析命令行参数：

import argparse

parser = argparse.ArgumentParser(description='这是一个演示argparse的脚本')

parser.add_argument('-n', '--name', type=str, help='姓名')
parser.add_argument('-a', '--age', type=int, help='年龄')
parser.add_argument('-g', '--gender', type=str, choices=['男', '女'], help='性别')

args = parser.parse_args()

print("你好，我的名字是{}，我今年{}岁，我的性别是{}".format(args.name, args.age, args.gender))

我们可以通过以下方式将参数传递给脚本：

python script.py -n Tom -a 18 -g 男

以上命令将在脚本中显示以下结果：

你好，我的名字是Tom，我今年18岁，我的性别是男

总结

在Python中，接受脚本输入是一项常见的任务。我们可以通过sys.argv获取命令行参数，也可以使用argparse解析命令行参数。对于较大的项目，argparse通常具有更好的可读性和维护性。