一个计算机技术爱好者与学习者

0%

好好学Python:Python获取命令行参数

1. Python获取命令行参数概述

在执行Python脚本时,很多时候需要从命令行获取用户输入的参数。

Pthon获取命令行参数的方式有多种,具体包括:

  • sys.argv
  • getopt
  • argparse
  • click

个人感觉其中最简单的是sys.argv,既简单又优雅的是argparse。本文,我们就来学习一下argparse的使用。

参考文档:

2. argparse示例

1、编写 test.py,内容如下

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
#!/usr/bin/env python
# -*- coding: utf-8 -*-

import argparse

# construct the argument parse and parse the arguments
ap = argparse.ArgumentParser()
ap.add_argument("-i", "--input", required=True, help="input filename")
ap.add_argument("-o", "--output", default='output.txt', help="output filename")
print(ap.parse_args())
print(vars(ap.parse_args()))
args = ap.parse_args()

print(f'input: {args.input}')
print(f'output: {args.output}')

2、测试执行

1
2
3
python test.py
python test.py -i input.txt -o output.txt
python test.py -i input.txt