提示词

这段 Python 代码因为有 input() 导致你在后台运行时报 EOFError 并且乱码。请你帮我修改这段代码，去掉 input()，改为使用 sys.argv 或 argparse 从命令行接收参数 (1 或 2)，并在顶部声明 # -- coding: utf-8 -- 解决乱码。