如何在Python中使用 sys.argv

你是否曾经需要在命令行中运行Python脚本并传递参数？sys.argv是实现这个过程的标准Python库。本文将介绍sys.argv的用法，我们将会探讨它如何工作，并展示一些简单的例子。

sys.argv是什么？

sys.argv是命令行参数的列表。命令行参数是当一个脚本从命令行调用时传递给它的参数。sys.argv的第一个参数始终是脚本的名称，即使在不同的环境中执行该脚本时，它也始终相同。在Python中，sys.argv是sys模块的一部分。

如何使用sys.argv

1. 获得命令行参数

首先，我们来看看如何获取命令行参数。下面是一个简单的Python脚本，它接受两个参数并将它们相加：

import sys

a = int(sys.argv[1])
b = int(sys.argv[2])

c = a + b

print(c)

运行脚本时，需要在命令行中传递两个数字来执行它：

$ python add.py 2 3
5

sys.argv[1]和sys.argv[2]分别是要加的两个数字。

2. 检查命令行参数的数量

如果脚本需要一个特定数量的参数，我们可以使用len(sys.argv)与所需参数数目进行比较。下面是一个例子：

import sys

if len(sys.argv) != 3:
    print("Usage: python add.py number1 number2")
    sys.exit(1)

a = int(sys.argv[1])
b = int(sys.argv[2])

c = a + b

print(c)

如果命令中传递少于或多于两个参数，则会打印一条消息，脚本将以状态值1而非0退出。

3. 处理多个命令行参数

如果需要处理多个命令行参数，我们可以使用一个循环来获取它们。下面是一个简单的例子，它将多个文件合并成一个文件：

import sys

if len(sys.argv) < 3:
    print("Usage: python merge.py output_file input_files...")
    sys.exit(1)

output_file = open(sys.argv[1], "w")

for input_file in sys.argv[2:]:
    with open(input_file) as f:
        output_file.write(f.read())

output_file.close()

该脚本需要至少传递两个参数：输出文件的名称和一个或多个输入文件的名称。它使用循环来处理所有输入文件，将它们的内容写入输出文件。这里使用了一个文件上下文管理器(with)，以确保文件被正确关闭。

总结

这篇文章提供了sys.argv的基础用法，高效获得并解析命令行参数。使用sys.argv，你可以让你的Python程序在命令行上更具交互性。但是，它的使用需要谨慎。在使用脚本时，务必小心一些错误，比如名字写错、参数传递错误等等.

参考资料

Python官方文档