Python 命令行参数|极客教程

Python 命令行参数

1. 前言

在编写 Python 程序时，有时候需要从命令行接收一些参数来控制程序的行为。Python 提供了 sys 模块用于处理命令行参数，我们可以使用该模块来获取命令行参数，并对其进行解析和处理。

本文将详细讲解如何在 Python 中使用命令行参数，并给出一些常用的示例。

2. sys 模块

sys 是 Python 标准库中的一个模块，提供了与解释器和操作系统进行交互的一些函数和变量。其中，sys.argv 是一个包含命令行参数的列表，它至少包含一个元素，即脚本文件的名称。其他的元素是从命令行输入的参数，按顺序依次存储在列表中。

接下来，我们将通过一些示例来介绍 sys.argv 的使用方法。

2.1 示例 1：输出命令行参数

下面是一个简单的示例，演示了如何使用 sys.argv 来输出命令行参数：

import sys

# 使用 sys.argv 输出命令行参数
for arg in sys.argv:
    print(arg)

运行以上代码并向其传递一些参数，例如 python script.py hello world，将会输出以下结果：

script.py
hello
world

在命令行中，我们可以通过空格将不同的参数分隔开。

2.2 示例 2：获取命令行参数

除了输出命令行参数，我们还可以根据实际需要获取其中的特定参数。

考虑以下示例代码，它从命令行参数中获取两个数字，并计算它们的和：

import sys

# 获取命令行参数并计算和
num1 = int(sys.argv[1])
num2 = int(sys.argv[2])
sum = num1 + num2

print(f"The sum of {num1} and {num2} is {sum}.")

我们可以在命令行中执行以下命令并观察输出：

python script.py 5 3

输出将会是：

The sum of 5 and 3 is 8.

在这个示例中，我们使用 int() 函数将获取的命令行参数转换为整数，然后进行相加操作。

需要注意的是，我们使用 sys.argv[1] 和 sys.argv[2] 来获取命令行参数。这是因为 sys.argv 的第一个元素是脚本文件的名称，后面才是实际的参数。

2.3 示例 3：处理可选参数

有时候，我们希望程序能够接受一些可选参数，以便根据用户的需求进行相应的操作。在这种情况下，我们可以使用 argparse 模块来处理命令行参数。

考虑以下示例代码，它使用 argparse 模块来接受一个可选的 --verbose 参数，并根据该参数的值输出不同级别的日志信息：

import argparse

# 创建 ArgumentParser 对象
parser = argparse.ArgumentParser()

# 添加 --verbose 参数
parser.add_argument("--verbose", help="increase output verbosity", action="store_true")

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

# 根据 --verbose 参数输出不同级别的日志信息
if args.verbose:
    print("Verbose output enabled")
    # 执行一些额外的操作
else:
    print("Only basic output")

如果我们在命令行中执行以下命令：