用户输入和命令行参数
-
09-06-2019 - |
题
如何拥有一个 Python 脚本,a) 可以接受用户输入,以及如何使其 b) 在从命令行运行时读取参数?
解决方案
要读取用户输入,您可以尝试 这 cmd
模块 用于轻松创建迷你命令行解释器(带有帮助文本和自动完成功能)以及 raw_input
(input
对于 Python 3+)用于读取用户的一行文本。
text = raw_input("prompt") # Python 2
text = input("prompt") # Python 3
命令行输入位于 sys.argv
. 。在你的脚本中尝试一下:
import sys
print (sys.argv)
有两个用于解析命令行选项的模块: (自 Python 2.7 起已弃用,使用 optparse
argparse
相反)和 getopt
. 。如果您只想将文件输入到脚本中,请看看 fileinput
.
这 Python 库参考 是你的朋友。
其他提示
var = raw_input("Please enter something: ")
print "you entered", var
或者对于 Python 3:
var = input("Please enter something: ")
print("You entered: " + var)
raw_input
在 Python 3.x 中不再可用。但 raw_input
被重命名 input
, ,因此存在相同的功能。
input_var = input("Enter something: ")
print ("you entered " + input_var)
处理命令行参数的最佳方法是 argparse
模块。
使用 raw_input()
获取用户输入。如果您导入 readline module
您的用户将拥有行编辑和历史记录。
小心不要使用 input
函数,除非你知道自己在做什么。不像 raw_input
, input
会接受任何Python表达式,所以它有点像 eval
这个简单的程序可以帮助您了解如何从命令行提供用户输入并显示有关传递无效参数的帮助。
import argparse
import sys
try:
parser = argparse.ArgumentParser()
parser.add_argument("square", help="display a square of a given number",
type=int)
args = parser.parse_args()
#print the square of user input from cmd line.
print args.square**2
#print all the sys argument passed from cmd line including the program name.
print sys.argv
#print the second argument passed from cmd line; Note it starts from ZERO
print sys.argv[1]
except:
e = sys.exc_info()[0]
print e
1) 求5的平方根
C:\Users\Desktop>python -i emp.py 5
25
['emp.py', '5']
5
2) 传递除数字之外的无效参数
C:\Users\bgh37516\Desktop>python -i emp.py five
usage: emp.py [-h] square
emp.py: error: argument square: invalid int value: 'five'
<type 'exceptions.SystemExit'>
使用“raw_input”从控制台/终端输入。
如果您只想要一个命令行参数,例如文件名或其他内容,例如
$ python my_prog.py file_name.txt
那么你可以使用 sys.argv...
import sys
print sys.argv
sys.argv 是一个列表,其中 0 是程序名称,因此在上面的示例中 sys.argv[1] 将是“file_name.txt”
如果您想拥有完整的命令行选项,请使用 optparse 模块。
佩夫
如果您运行的是 Python <2.7,则需要 优化分析, ,正如文档所解释的,它将创建一个在应用程序运行时调用的命令行参数的接口。
然而,在Python ≥2.7中,optparse已被弃用,并被替换为 arg解析 如上图所示。文档中的一个快速示例...
以下代码是一个Python程序,该程序列出了整数列表并产生总和或最大:
import argparse
parser = argparse.ArgumentParser(description='Process some integers.')
parser.add_argument('integers', metavar='N', type=int, nargs='+',
help='an integer for the accumulator')
parser.add_argument('--sum', dest='accumulate', action='store_const',
const=sum, default=max,
help='sum the integers (default: find the max)')
args = parser.parse_args()
print args.accumulate(args.integers)
从Python开始 3.2 2.7,现在有 arg解析 用于处理命令行参数。
如果是 3.x 版本,则只需使用:
variantname = input()
例如,您要输入8:
x = input()
8
x 将等于 8,但它将是一个字符串,除非您以其他方式定义它。
所以你可以使用转换命令,例如:
a = int(x) * 1.1343
print(round(a, 2)) # '9.07'
9.07
在Python 2中:
data = raw_input('Enter something: ')
print data
在Python 3中:
data = input('Enter something: ')
print(data)
import six
if six.PY2:
input = raw_input
print(input("What's your name? "))