如果文件不超过n行,请更改变量

时间:2018-12-11 21:32:34

标签: python

我目前有一些代码可以捕获文件中第一行之后的所有行,并将其保存到变量 resourceslist 中。我想添加一些代码,说明如果文件中只有 行,然后为变量 resourceslist 提供值“ oneline”

with open('filepaths', "r+") as f:
            if index + 1 > len(f):
                for _ in range(1):
                    next(f)
                for lines in f:
                    resourceslist = f.read()
               else:
                   resourceslist = "oneline"

1 个答案:

答案 0 :(得分:1)

您可以编写以下内容;您不需要进行第一个for循环,因为它实际上不会循环,而第二个循环是不必要的,因为您希望将文件的全部(剩余)内容读入resourceslist中,而无需进行其他迭代在剩余的行上。

with open('filepath') as f:
    next(f)  # Skip the first line
    resourceslist = f.read()
    if not resourceslist:  # i.e., f.read() returned the empty string
        resourceslist = "oneline"