带枚举的基本python file-io变量

python的新手,并试图学习文件i / o的绳索.

我正在使用以下格式从大型(200万行)文件中提取行:

56fr4
4543d
4343d
5irh3

这是我用来返回代码的函数:

def getCode(i):
    with open("test.txt") as f:
        for index, line in enumerate(f):
            if index == i:
                code = # what does it equal?
                break
    return code

一旦索引到达正确的位置(i),我使用什么语法来设置代码变量?

解决方法:

code = line.strip()

将删除尾随的新行时,将代码分配给等于i的行号.

您还需要稍微更新您的代码

 def getCode(i):
    with open('temp.txt', 'r') as f:
             for index, line in enumerate(f):
                     if index == i:
                             code = line.strip()
                             return code

为什么你需要.strip()

>>> def getCode(i):
...     with open('temp.txt') as f:
...             for index, line in enumerate(f):
...                     if index == i:
...                             code = line
...                             return code
 ... 
>>> getCode(2)
"                  'LINGUISTIC AFFILIATION',\n"

是的,“’LINGUISTIC AFFILIATION’,”是我目前的temp.txt’

上一篇:leetcode-5394-对角线遍历②


下一篇:使用缓冲读取器来处理大型.csv文件,Python