Python命令行參數

Python是一種流行的面向對象編程語言，其特點之一是靈活性。有時開發人員需要從命令行接收外部輸入。Python可以通過命令行參數來實現這一目的。命令行是一個在終端或控制台中輸入的文本行，Python解釋器可以讀取並解析這些命令行參數。在本文中，我們將探討Python中的命令行參數以及如何使用它們進行編程。

一、傳遞命令行參數

在Python中，命令行參數通過sys模塊中的argv函數傳遞。argv函數返回一個列表，其中包含從命令行傳遞給Python腳本的所有參數。例如，以下是一個Python腳本，用於將從命令行接收的參數相加：

import sys

sum = 0
for arg in sys.argv:
    sum += int(arg)

print("Sum is:", sum)

在終端或控制台中，通過輸入以下命令運行腳本並傳遞參數：

python script.py 4 6 8

該腳本將打印出“Sum is: 18”。

二、解析命令行參數

有時候，傳遞參數列表並不能滿足我們的需求。我們需要更靈活的方式來解析並使用命令行參數。Python提供了argparse模塊來實現這個目標。該模塊可以幫助我們定義腳本所需的參數並解析它們。

以下示例演示了如何使用argparse模塊來接收和解析命令行參數：

import argparse

parser = argparse.ArgumentParser()
parser.add_argument("--name", help="the name to greet")
parser.add_argument("--greeting", help="the greeting to use")

args = parser.parse_args()

print(args.name + ", " + args.greeting)

在終端或控制台中，通過輸入以下命令運行腳本並傳遞參數：

python script.py --name Alice --greeting "Hello!"

該腳本將打印出“Alice, Hello!”。

三、結合默認參數的命令行參數

有時候，我們想要在定義命令行參數時設置默認值。在這種情況下，我們可以使用默認參數值。以下示例演示了如何使用argparse模塊結合默認參數值來接收和解析命令行參數：

import argparse

parser = argparse.ArgumentParser()
parser.add_argument("--name", help="the name to greet", default="Alice")
parser.add_argument("--greeting", help="the greeting to use", default="Hello!")

args = parser.parse_args()

print(args.name + ", " + args.greeting)

在終端或控制台中，通過輸入以下命令運行腳本，並省略參數：

python script.py

該腳本將打印出“Alice, Hello!”。

四、使用只有選項參數的命令行參數

選項參數是一個開關，表示某些函數的開啟或關閉。它是一個有助於定義負載的極好邏輯，有很多Python庫可以使用它來增加它們的功能。argparse模塊允許我們定義這樣的選項參數。例如，以下是將一條數據寫入CSV文件中的腳本，其中使用了選項參數：“-a”表示用追加方式寫入，而“-n”表示在寫入前不創建文件：

import argparse
import csv

parser = argparse.ArgumentParser()
parser.add_argument("-a", "--append", help="append to file", action="store_true")
parser.add_argument("-n", "--newfile", help="don't create file", action="store_true")
parser.add_argument("data", help="data to write to csv")

args = parser.parse_args()

mode = 'a' if args.append else 'w'
if not args.newfile:
    with open('data.csv', mode) as f:
        writer = csv.writer(f)
        writer.writerow([args.data])
else:
    print(args.data)

在終端或控制台中，通過輸入以下命令運行腳本，並帶有選項參數：

python script.py --append --data "Hello, world!"

該腳本將把“Hello, world!”寫入文件data.csv中。

五、使用位置參數的命令行參數

位置參數是指按參數順序傳遞的參數，而不是通過關鍵字傳遞的命名參數。例如，在以下Python腳本中，位置參數被解析為輸入文件名和輸出文件名：

import argparse

parser = argparse.ArgumentParser()
parser.add_argument("input_file", help="input filename")
parser.add_argument("output_file", help="output filename")

args = parser.parse_args()

print("Input file:", args.input_file)
print("Output file:", args.output_file)

在終端或控制台中，通過輸入以下命令運行腳本，並帶有位置參數：

python script.py input.txt output.txt

該腳本將打印出“Input file: input.txt”和“Output file: output.txt”。

總結：

Python命令行參數是一種方便的外部輸入方式，可以實現程序的靈活性。在Python中，我們可以使用sys模塊或argparse模塊來解析和使用命令行參數。argparse模塊除了支持傳遞參數列表外，還支持選項參數和位置參數，並且還可以與默認參數值結合使用。通過熟練使用Python命令行參數，可以提高程序的交互性和靈活性。