是的,Python命令行参数可以自定义
import sys
def main(arg1, arg2):
print(f"参数1: {arg1}")
print(f"参数2: {arg2}")
if __name__ == "__main__":
# 获取命令行参数,第一个参数是脚本名称,所以从索引1开始
arg1 = sys.argv[1]
arg2 = sys.argv[2]
main(arg1, arg2)
在这个例子中,我们定义了一个名为main
的函数,它接受两个参数。在if __name__ == "__main__"
代码块中,我们从sys.argv
列表中获取命令行参数。sys.argv[0]
是脚本名称,所以我们从索引1开始获取参数。
现在,当你运行这个脚本并提供命令行参数时,例如:
python script.py 参数1 参数2
它将输出:
参数1: 参数1
参数2: 参数2
你可以根据需要自定义这些参数。