跳转至内容

Python 编程/输入和输出

来自 Wikibooks,开放世界中的开放书籍

Python 3.x 只有一个函数用于获取用户输入,即 input()。相比之下,旧版本的 Python 2.x 有两个函数用于获取用户输入:input()raw_input()

还有一些非常简单的方法可以读取文件,以及在需要更严格控制输入的情况下,从标准输入读取。

Python 3.x 中的 input()

[编辑 | 编辑源代码]

在 Python 3.x 中,input() 会提示用户输入一个字符串(以换行符结束),并简单地返回该字符串。它还可以接受一个参数,该参数会在用户输入数据之前显示为提示。例如:

print(input('What is your name?'))

输出

What is your name? <user input data here>

示例:将用户输入的姓名(即字符串数据)赋值给变量“x”,您需要输入

x = input('What is your name?')

在旧版本的 Python 2.x 中,上述内容适用于 raw_input() 函数,并且还有一个 input() 函数的行为有所不同,会自动评估用户输入的内容;在 Python 3 中,可以通过 eval(input()) 实现相同的功能。

链接

  • input() 在 Python 3 库参考中的内置函数中,docs.python.org
  • raw_input() 在 Python 2 库参考中的内置函数中,docs.python.org

Python 2.x 中的 input()

[编辑 | 编辑源代码]

在旧版本的 Python 2.x 中,input() 将用户输入作为字符串并对其进行评估。

因此,如果脚本说

x = input('What are the first 10 perfect squares? ')

用户可以输入

map(lambda x: x*x, range(10))

这将以列表形式给出正确的答案。请注意,任何输入的语句都无法跨越多行。

出于安全原因,input() 不应用于除最简单的程序之外的任何程序。使用以下习惯用法将 raw_input() 返回的字符串转换为 Python 类型

x = None
while not x:
    try:
        x = int(raw_input())
    except ValueError:
        print('Invalid Number')

更可取,因为 input() 使用 eval() 将文字转换为 Python 类型,这使得恶意人员可以轻松地从您的程序内部运行任意代码。

链接

  • input() 在 Python 2 库参考中的内置函数中,docs.python.org

文件输入

[编辑 | 编辑源代码]

文件对象

[编辑 | 编辑源代码]

要从文件读取,可以使用 open 迭代文件的每一行

f = open('test.txt', 'r')
for line in f:
    print(line[0])
f.close()

这将打印每一行的第一个字符。换行符附加在以这种方式读取的每一行的末尾。open 的第二个参数可以是 'r'、'w' 或 'rw',以及其他一些参数。

读取文件的新方法且更好

with open("test.txt", "r") as txt:
    for line in txt:
        print(line)

优点是,在完成 with 语句内的部分后,打开的文件将自动关闭,即使抛出异常也会如此。

因为当文件对象超出范围时,文件会自动关闭,所以没有必要显式关闭它们。因此,前面代码中的循环也可以写成

for line in open('test.txt', 'r'):
    print(line[0])

您可以一次读取特定数量的字符

c = f.read(1)
while len(c) > 0:
    if len(c.strip()) > 0: print(c)
    c = f.read(1)

这将一次从 f 读取一个字符,然后在它们不是空格时打印它们。

文件对象隐式包含一个标记来表示当前位置。如果应将文件标记移回开头,则可以关闭文件对象并重新打开它,或者只需使用以下命令将标记移回开头:

f.seek(0)

标准文件对象

[编辑 | 编辑源代码]

有一些内置文件对象表示标准输入、输出和错误。这些位于 sys 模块中,分别称为 stdin、stdout 和 stderr。在 __stdin__、__stdout__ 和 __stderr__ 中也有这些对象的不可变副本。这是为了 IDLE 和其他标准文件已被更改的工具。

您必须导入 sys 模块才能使用特殊的 stdin、stdout、stderr I/O 处理程序。

import sys

为了更精细地控制输入,请使用 sys.stdin.read()。要在 Python 中实现 UNIX 的“cat”程序,您可以执行以下操作

import sys
for line in sys.stdin:
    print(line, end="")

请注意,sys.stdin.read() 将从标准输入读取到 EOF。(通常是 Ctrl+D。)

解析命令行

[编辑 | 编辑源代码]

传递给 Python 程序的命令行参数存储在 sys.argv 列表中。列表中的第一个项目是 Python 程序的名称,它可能包含也可能不包含完整路径,具体取决于调用方式。sys.argv 列表是可修改的。

打印所有传递的参数,除了程序名称本身

import sys
for arg in sys.argv[1:]:
  print(arg)

解析传递的参数以获取传递的减号选项

import sys
option_f = False
option_p = False
option_p_argument = ""
i = 1
while i < len(sys.argv):
  if sys.argv[i] == "-f":
    option_f = True
    sys.argv.pop(i)
  elif sys.argv[i] == "-p":
    option_p = True
    sys.argv.pop(i)
    option_p_argument = sys.argv.pop(i)
  else:
    i += 1

在上面,找到选项的参数被删除,以便可以循环遍历 sys.argv 以获取所有剩余的参数。

命令行参数的解析还得到库模块 optparse(已弃用)、argparse(从 Python 2.7 开始)和 getopt(为了方便 C 程序员)的支持。

argparse 的最小解析示例

import argparse
parser = argparse.ArgumentParser(description="Concatenates two strings")
addarg = parser.add_argument
addarg("s1", help="First string to concatenate")
addarg("s2", help="Second string to concatenate")
args = parser.parse_args()
result = args.s1 + args.s2
print(result)

使用 argparse 解析——将 arg 类型指定为 int

import argparse
parser = argparse.ArgumentParser(description="Sum two ints")
addarg = parser.add_argument
addarg("i1", help="First int to add", type=int)
addarg("i2", help="Second int to add", type=int)
args = parser.parse_args()
result = args.i1 + args.i2
print(result)

使用 argparse 解析——添加可选开关 -m 以产生乘法而不是加法

import argparse
parser = argparse.ArgumentParser(description="Sums or multiplies two ints.")
addarg = parser.add_argument
addarg("i1", help="First int", type=int)
addarg("i2", help="Second int", type=int)
addarg("-m", help="Multiplies rather than adds.", action="store_true")
args = parser.parse_args()
if args.m:
  result = args.i1 * args.i2
else:
  result = args.i1 + args.i2
print(result)

使用 argparse 解析——将参数设置为使用一个或多个项目

import argparse
parser = argparse.ArgumentParser(description="Sums one or more ints.")
addarg = parser.add_argument
addarg("intlist", help="Ints", type=int, nargs="+")
args = parser.parse_args()
result = 0
for item in args.intlist:
  result += item
print(result)

使用示例:python ArgparseTest.py 1 3 5

使用 argparse 解析——与上面相同,但在参数描述后使用 -h 输出帮助结语

import argparse
parser = argparse.ArgumentParser(description="Sums one or more ints.",
  epilog="Example: python ArgparseTest.py 1 3 5")
addarg = parser.add_argument
addarg("intlist", help="Ints", type=int, nargs="+")
args = parser.parse_args()
result = 0
for item in args.intlist:
  result += item
print(result)

使用 argparse 解析——通过 nargs 使第二个整数参数可选

import argparse
parser = argparse.ArgumentParser(description="Sums one or two integers.",
  epilog="Example: python ArgparseTest.py 3 4\n"
         "Example: python ArgparseTest.py 3")
addarg = parser.add_argument
addarg("i1", help="First int", type=int)
addarg("i2", help="Second int, optional, defaulting to 1.", type=int, default=1, nargs="?")
args = parser.parse_args()
result = args.i1 + args.i2
print(result)

链接

进行输出的基本方法是 print 语句。

print('Hello, world')

要在同一行打印多个内容,并用空格分隔,请在它们之间使用逗号

print('Hello,', 'World')

这将打印以下内容

Hello, World

虽然这两个字符串都不包含空格,但由于两个对象之间的逗号,print 语句添加了一个空格。可以打印任意数据类型

print(1, 2, 0xff, 0777, 10+5j, -0.999, map, sys)

这将输出以下内容

1 2 255 511 (10+5j) -0.999 <built-in function map> <module 'sys' (built-in)>

可以在同一行打印对象,而无需在同一行上

for i in range(10):
    print(i, end=" ")

这将输出以下内容

0 1 2 3 4 5 6 7 8 9

要以换行符结束打印的行,请添加一个没有任何对象的 print 语句。

for i in range(10):
    print(i, end=" ")
print()
for i in range(10,20):
    print(i, end=" ")

这将输出以下内容

0 1 2 3 4 5 6 7 8 9
10 11 12 13 14 15 16 17 18 19

如果不存在空的 print 语句,则上述输出将如下所示

0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19

您可以打印到文件而不是标准输出

print('Hello, world', file=f)

这将打印到任何实现了 write() 的对象,包括文件对象。

关于旧版本的 Python 2 的说明:在 Python 2 中,print 是一个语句而不是函数,并且无需在其参数周围加上括号。而不是print(i, end=" "),您将编写print i,.

省略换行符

[编辑 | 编辑源代码]

在 Python 3.x 中,您可以通过将 end="" 传递给 print 函数或使用 write 方法来输出而无需换行符

import sys
print("Hello", end="")
sys.stdout.write("Hello") # Or stderr to write to standard error stream.

在 Python 2.x 中,要避免在对象的输出与随后的 print 语句之间添加空格和换行符,您可以执行以下操作之一

连接:连接每个对象的字符串表示形式,然后稍后一次打印整个内容。

print(str(1)+str(2)+str(0xff)+str(0777)+str(10+5j)+str(-0.999)+str(map)+str(sys))

这将输出以下内容

12255511(10+5j)-0.999<built-in function map><module 'sys' (built-in)>

Write 函数:您可以为 sys.stdout.write 创建一个简写,并将其用于输出。

import sys
write = sys.stdout.write
write('20')
write('05\n')

这将输出以下内容

2005

您可能需要 sys.stdout.flush() 以快速在屏幕上显示该文本。

Python 3.x 输出示例

  • from __future__ import print_function
    • 确保 Python 2.6 及更高版本的 Python 2.x 可以使用 Python 3.x 的 print 函数。
  • print("Hello", "world")

    • 打印两个单词,单词之间用空格隔开。注意周围的括号,在 Python 2.x 中未使用。
  • print("Hello world", end="")
    • 打印不带结尾换行符。
  • print("Hello", "world", sep="-")
    • 打印两个单词,单词之间用短横线隔开。
  • print("Hello", 34)
    • 打印各种数据类型的元素,用空格分隔。
  • print("Hello " + 34)
    • 由于尝试连接字符串和整数,导致抛出错误。
  • print("Hello " + str(34))
    • 使用 "+" 连接字符串,在将数字转换为字符串后。
  • sum=2+2; print "The sum: %i" % sum
    • 打印一个字符串,该字符串已使用作为参数传递的整数进行格式化。另请参阅 #格式化
  • print ("Error", file=sys.stderr)
    • 输出到文件句柄,在本例中为标准错误流。

Python 2.x 输出示例

  • print "Hello"
  • print "Hello", "world"
    • 用空格分隔两个单词。
  • print "Hello", 34
    • 打印各种数据类型的元素,用空格分隔。
  • print "Hello " + 34
    • 由于尝试连接字符串和整数,导致抛出错误。
  • print "Hello " + str(34)
    • 使用 "+" 连接字符串,在将数字转换为字符串后。
  • print "Hello",
    • 打印“Hello ”不带换行符,并在末尾带一个空格。
  • sys.stdout.write("Hello")
    • 打印“Hello”不带换行符。“import sys”是先决条件。需要后续的“sys.stdout.flush()”才能立即显示在用户的屏幕上。
  • sys.stdout.write("Hello\n")
    • 打印“Hello”带换行符。
  • print >> sys.stderr, "An error occurred."
    • 打印到标准错误流。
  • sys.stderr.write("Hello\n")
    • 打印到标准错误流。
  • sum=2+2; print "The sum: %i" % sum
    • 打印一个字符串,该字符串已使用作为参数传递的整数进行格式化。
  • formatted_string = "The sum: %i" % (2+2); print formatted_string
    • 与前面类似,只是格式化发生在 print 语句之外。
  • print "Float: %6.3f" % 1.23456
    • 输出“Float: 1.234”。句点后的数字 3 指定要显示的小数位数,而句点前的数字 6 指定显示的数字应占用的字符总数,如果需要则用空格填充。
  • print "%s is %i years old" % ("John", 23)
    • 将两个参数传递给格式化程序。

文件输出

[编辑 | 编辑源代码]

将数字 1 到 10 打印到文件,每行一个数字

file1 = open("TestFile.txt","w")
for i in range(1,10+1):
  print(i, file=file1)
file1.close()

使用“w”,文件将以写入模式打开。使用“file=file1”,print 将其输出发送到文件而不是标准输出。

将数字 1 到 10 打印到文件,用短横线分隔

file1 = open("TestFile.txt", "w")
for i in range(1, 10+1):
  if i > 1:
    file1.write("-")
  file1.write(str(i))
file1.close()

打开文件以追加而不是覆盖

file1 = open("TestFile.txt", "a")

在 Python 2.x 中,重定向到文件的方式如下print >>file1, i.

另请参阅 文件 章节。

格式化

[编辑 | 编辑源代码]

使用字符串百分号运算符将数字和其他值格式化为字符串

v1 = "Int: %i" % 4               # 4
v2 = "Int zero padded: %03i" % 4 # 004
v3 = "Int space padded: %3i" % 4 #   4
v4 = "Hex: %x" % 31              # 1f
v5 = "Hex 2: %X" % 31            # 1F - capitalized F
v6 = "Oct: %o" % 8               # 10
v7 = "Float: %f" % 2.4           # 2.400000
v8 = "Float: %.2f" % 2.4         # 2.40
v9 = "Float in exp: %e" % 2.4    # 2.400000e+00
vA = "Float in exp: %E" % 2.4    # 2.400000E+00
vB = "List as string: %s" % [1, 2, 3]
vC = "Left padded str: %10s" % "cat"
vD = "Right padded str: %-10s" % "cat"
vE = "Truncated str: %.2s" % "cat"
vF = "Dict value str: %(age)s" % {"age": 20}
vG = "Char: %c" % 65             # A
vH = "Char: %c" % "A"            # A

自 Python 2.6 起,使用 format() 字符串方法将数字和其他值格式化为字符串

v1 = "Arg 0: {0}".format(31)     # 31
v2 = "Args 0 and 1: {0}, {1}".format(31, 65)
v3 = "Args 0 and 1: {}, {}".format(31, 65)
v4 = "Arg indexed: {0[0]}".format(["e1", "e2"])
v5 = "Arg named: {a}".format(a=31)
v6 = "Hex: {0:x}".format(31)     # 1f
v7 = "Hex: {:x}".format(31)      # 1f - arg 0 is implied
v8 = "Char: {0:c}".format(65)    # A
v9 = "Hex: {:{h}}".format(31, h="x") # 1f - nested evaluation

自 Python 3.6 起,使用文字字符串插值将数字和其他值格式化为字符串

int1 = 31; int2 = 41; str1="aaa"; myhex = "x"
v1 = f"Two ints: {int1} {int2}"
v2 = f"Int plus 1: {int1+1}"      # 32 - expression evaluation
v3 = f"Str len: {len(str1)}"      # 3 - expression evaluation
v4 = f"Hex: {int1:x}"             # 1f
v5 = f"Hex: {int1:{myhex}}"       # 1f - nested evaluation

链接

[编辑 | 编辑源代码]
华夏公益教科书