获取参数函数python

当我们编写Python脚本时，有时需要从命令行或者其他途径获取一些参数来指定程序运行时的行为。Python提供了多种获取参数的方式，本文将介绍其中常用的几种方式。

1. 命令行参数

使用命令行参数是最常见的获取参数方式之一。在Python中，可以使用sys.argv获取命令行参数。sys.argv是一个列表，其中包含脚本名称和所有的参数。例如，下面这个简单的脚本可以获取命令行参数并打印：

import sys

print("脚本名：", sys.argv[0])
print("参数：", sys.argv[1:])

运行这个脚本并传入一些参数：

python myscript.py arg1 arg2 arg3

输出：

脚本名： myscript.py
参数： ['arg1', 'arg2', 'arg3']

2. argparse模块

argparse是Python自带的一个命令行参数解析模块，它能够自动生成帮助信息，并且能够根据我们定义的参数格式进行严格的参数解析。使用argparse可以让我们更轻松地编写命令行程序。

首先，我们需要导入argparse模块：

import argparse

接下来，我们可以定义一些命令行参数：

parser = argparse.ArgumentParser(description='描述参数用途')
parser.add_argument('arg1', help='参数1的帮助信息')
parser.add_argument('arg2', help='参数2的帮助信息')
parser.add_argument('-v', '--verbose', help='增加输出信息', action='store_true')

这段代码定义了三个参数，arg1和arg2是必选参数，-v或者--verbose是可选参数。action='store_true'表示如果传入了-v参数，则args.verbose的值为True，否则为False。

定义好参数之后，我们可以使用parse_args()方法解析命令行参数：

args = parser.parse_args()

这个方法会返回一个命名空间对象args，我们可以通过args.arg1、args.arg2和args.verbose来访问对应的命令行参数值。

下面是一个完整的范例：

import argparse

parser = argparse.ArgumentParser(description='这是一个示例程序')
parser.add_argument('arg1', help='参数1的帮助信息')
parser.add_argument('arg2', help='参数2的帮助信息')
parser.add_argument('-v', '--verbose', help='增加输出信息', action='store_true')

args = parser.parse_args()

print("arg1:", args.arg1)
print("arg2:", args.arg2)
print("verbose:", args.verbose)

3. configparser模块

configparser是Python自带的一个INI文件解析模块，它可以很方便地从INI配置文件中读取参数。

首先，我们需要导入configparser模块：

import configparser

然后，我们可以创建一个ConfigParser对象并读取一个INI文件：

config = configparser.ConfigParser()
config.read('config.ini')

下面是一个简单的INI配置文件：

[DATABASE]
HOST = localhost
PORT = 5432
USER = postgres
PASSWORD = p@ssword

我们可以使用get()方法获取对应的配置项：

host = config.get('DATABASE', 'HOST')
port = config.getint('DATABASE', 'PORT')
user = config.get('DATABASE', 'USER')
password = config.get('DATABASE', 'PASSWORD')

这里使用了getint()方法来获取整数类型的配置项。

完整范例：

import configparser

config = configparser.ConfigParser()
config.read('config.ini')

host = config.get('DATABASE', 'HOST')
port = config.getint('DATABASE', 'PORT')
user = config.get('DATABASE', 'USER')
password = config.get('DATABASE', 'PASSWORD')

print("HOST:", host)
print("PORT:", port)
print("USER:", user)
print("PASSWORD:", password)

结论

以上就是Python中常用的获取参数的方式。使用命令行参数、argparse模块和configparser模块可以很方便地从外部获取参数，让我们的程序变得更加灵活和可配置。