Python3多字符串打印下一步

Posted 2021-05-05

如何使用next（）函数打印以“string”开头的每行下面的行？我只找到了如何用.txt文档做的答案，但在这种情况下我不想使用文本文档。以下是我迄今为止尝试过的内容......

longString = """
    string1
    01

    string2
    02

    string3
    03
    """
lines = longString.split('
')
for line in lines:
    if line.startswith(str(mylist[0])): # The mylist[0] is "string"
        print(line)

Output:
string1
string2
string3

答案

longString = """
string1
01

string2
02

string3
03
"""
lines = iter(longString.split('
'))
for line in lines:
    if line.startswith("string"):
        print(next(lines))

Output:
01
02
03

你需要一个实际的迭代器（不只是一个可迭代的序列）来使用next() - 而内置的iter()函数就是这样做的。

请注意，我必须删除字符串中的前导空格 - 没有一行实际上以“string”的方式开始。