如何从输出字符串中搜索指定行并打印其后的两行内容？

嘿，我懂你想要的效果了——原来的代码是匹配到以----开头的行后，手动输出固定的两行内容，现在你希望改成自动抓取该行之后的恰好两行内容，对吧？这不难实现，我们可以通过追踪行的索引来定位后续内容，同时还要注意避免索引越界的问题（比如如果匹配的行是最后一行，就不会报错）。

先回顾下你原来的代码：

a = """
Some lines I do not want
-----
--------
--
I need this line
I need this line also
Again few lines i do not want
"""
for line in a.split("\n"):
    if line.startswith("----"):
        print "I need this line"
        print "I need this line also"

下面是修改后的代码，完全满足你的需求：

a = """
Some lines I do not want
-----
--------
--
I need this line
I need this line also
Again few lines i do not want
"""

# 将字符串分割为行的列表，保留原有的行结构
lines = a.split("\n")

# 使用enumerate同时获取行的索引和内容
for idx, line in enumerate(lines):
    # 匹配以"----"开头的行
    if line.startswith("----"):
        # 先检查后续是否有足够的行（至少两行）
        if idx + 2 < len(lines):
            print(lines[idx + 1])
            print(lines[idx + 2])
        # 如果只有一行后续内容，也可以选择打印（可选逻辑，根据你需求调整）
        elif idx + 1 < len(lines):
            print(lines[idx + 1])
        # 如果没有后续内容，就什么都不做，避免报错

代码逻辑解释： #

• 用enumerate()遍历行列表，这样我们能同时拿到每行的索引和内容，这是实现定位后续行的关键。
• 当匹配到目标行时，通过idx+1和idx+2直接取到后面的两行内容，代替原来的手动打印。
• 增加了索引检查：idx + 2 < len(lines)确保我们不会去访问列表不存在的索引，避免抛出IndexError。

这样修改后，代码就会自动输出匹配行之后的两行内容，完全符合你的需求啦！

内容的提问来源于stack exchange，提问作者Sagar