Python中如何定位文本文件指定行并查看目标行后两行内容

Got it, let's break down how to solve this problem—you're already using enumerate() to read lines with their numbers, and now you need to peek at the next two lines once you find your target string. Here are two solid approaches depending on your file size:

Method 1: Read All Lines Into a List (Best for Small/Medium Files) #

This is the most straightforward way. By storing all lines in a list first, you can directly access any line using its index (which matches the line_number from enumerate()).

# First, read all lines from the file into a list
with open("your_file.txt", "r") as file:
    lines = file.readlines()

# Iterate through each line with its number
for line_number, line in enumerate(lines):
    # Check if current line has your first target string
    if "specific_string_1" in line:
        # Make sure there are at least two lines left to check
        if line_number + 2 < len(lines):
            # Grab the next two lines using their indices
            next_line = lines[line_number + 1]
            next_next_line = lines[line_number + 2]
            
            # Check if either of those lines has your second target string
            if "specific_string_2" in next_line or "specific_string_2" in next_next_line:
                print(f"Found 'specific_string_2' in lines {line_number + 2} or {line_number + 3}!")
        else:
            print("Not enough lines remaining after the target line to check.")

Why this works: #

• Lists let you jump directly to any line index, so once you have line_number, you just add 1 and 2 to get the next two lines.
• You avoid issues with file iteration state, since everything is stored in memory.
• Always add the check line_number + 2 < len(lines) to prevent an IndexError if your target string is near the end of the file.

Method 2: Use an Iterator for Large Files (Memory-Friendly) #

If you're working with a huge file that can't fit into memory, you can use a file iterator to manually fetch the next lines without loading everything at once.

with open("your_file.txt", "r") as file:
    # Convert the file object into an iterator to control reads manually
    line_iterator = iter(file)
    
    for line_number, line in enumerate(line_iterator):
        if "specific_string_1" in line:
            try:
                # Fetch the next line from the iterator
                next_line = next(line_iterator)
                # Fetch the line after that
                next_next_line = next(line_iterator)
                
                if "specific_string_2" in next_line or "specific_string_2" in next_next_line:
                    print(f"Found 'specific_string_2' in lines {line_number + 2} or {line_number + 3}!")
            except StopIteration:
                # This exception is raised if there are no more lines to read
                print("Not enough lines left after the target line.")

Note for this method: #

• When you call next(line_iterator), you're advancing the iterator past those lines, so the enumerate() loop will skip them in its next iteration. If you need to continue processing all lines after this, you'd have to handle that (but usually, if you're only checking for this specific pattern, it's not an issue).
• The try-except block catches the case where there aren't two lines left after your target.

Key Takeaways #

• For most cases, Method 1 is simpler and easier to debug.
• Use Method 2 only when dealing with files too large to fit into memory.
• Always handle edge cases where the target string is near the end of the file to avoid crashes.

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