将文本文件中的行读入变量

2024-05-16 09:47:47 发布

您现在位置:Python中文网/ 问答频道 /正文

我的程序中有两个不同的函数,一个将输出写入txt文件(函数a),另一个读取并将其用作输入(函数B)。

函数A工作得很好(尽管我总是乐于接受关于如何改进的建议)。 看起来是这样的:

def createFile():
    fileName = raw_input("Filename: ")
    fileNameExt = fileName + ".txt" #to make sure a .txt extension is used

    line1 = "1.1.1"
    line2 = int(input("Enter line 2: ")
    line3 = int(input("Enter line 3: ")

    file = (fileNameExt, "w+")
    file.write("%s\n%s\n%s" % (line1, line2, line3))
    file.close()

    return

这看起来工作正常,将创建一个类似

1.1.1
123
456

现在,函数B应该使用该文件作为输入。这就是我到目前为止所取得的成就:

def loadFile():
    loadFileName = raw_input("Filename: ")
    loadFile = open(loadFileName, "r")

    line1 = loadFile.read(5)

这就是我陷入困境的地方,我知道如何使用前5个字符,但我也需要第2行和第3行作为变量。


Tags: 文件函数txtinputrawdeffilenamefile
2条回答
f = open('file.txt')
lines = f.readlines()
f.close()

lines就是你想要的

其他选择:

f = open( "file.txt", "r" )
lines = []
for line in f:
    lines.append(line)
f.close()

更多阅读:

https://docs.python.org/2/tutorial/inputoutput.html#reading-and-writing-files

from string import ascii_uppercase
my_data = dict(zip(ascii_uppercase,open("some_file_to_read.txt"))
print my_data["A"]

这将把它们存储在字典中,并以出租者为键。。。如果你真的想把它塞进变量中(注意,通常这是一个糟糕的想法),你可以这样做

globals().update(my_data)
print A

相关问题 更多 >