在Python脚本中设置参数可以通过以下几种方法:
1. 使用`input`函数:
def test(a, b):
print(f"{a} + {b} = {a + b}")
if __name__ == "__main__":
a = int(input("输入第一个加数:"))
b = int(input("输入第二个加数:"))
test(a, b)
2. 使用`argparse`库:
import argparse
parser = argparse.ArgumentParser(description="一个简单的加法程序")
parser.add_argument("a", type=int, help="第一个加数")
parser.add_argument("b", type=int, help="第二个加数")
def test(a, b):
print(f"{a} + {b} = {a + b}")
if __name__ == "__main__":
args = parser.parse_args()
test(args.a, args.b)
3. 使用`sys.argv`:
import sys
def test(a, b):
print(f"{a} + {b} = {a + b}")
if __name__ == "__main__":
a = int(sys.argv)
b = int(sys.argv)
test(a, b)
4. 使用第三方库,如`click`:
import click
@click.command()
@click.argument("a", type=int)
@click.argument("b", type=int)
def test(a, b):
"""简单的加法程序"""
click.echo(f"{a} + {b} = {a + b}")
if __name__ == "__main__":
test()
使用`argparse`库的好处是可以通过命令行参数直接运行脚本,并且可以方便地提供帮助信息和版本信息:
python script.py --help 显示帮助信息
python script.py -a 1 -b 2 运行脚本并传递参数
python script.py --version 显示版本信息
请根据您的需求选择合适的方法来设置参数