Python如何跳过文件前若干行直至找到指定前缀行（无需编写比较函数）

我需要读取如下格式的文本文件：

junk_line_1 
junk_line_2
junk_line_3
real data starts here

需求是跳过所有行，直至找到以指定前缀（比如'real data'）开头的行。

目前已实现的解决方案如下：

import itertools

def has_prefix(line):
  return not line.startswith('real data')

def read_file(f):
  lines=[]
  for line in itertools.dropwhile(has_prefix, f):
    lines.append(line.rstrip())
  return lines

但我不想单独写这个has_prefix函数：一是写法不够简洁，二是它只能支持固定前缀。我原本想写成这样（但语法不正确）：

import itertools

def read_file(f, prefix):
  lines=[]
  for line in itertools.dropwhile(not line.startswith(prefix), f) :
    lines.append(line.rstrip())
  return lines

由于line在循环开始前未定义，这种写法不可行。有没有更优雅的实现方式？至少要能把前缀作为参数传入，而不是硬编码。

1. 使用Lambda表达式替代单独函数 #

直接在itertools.dropwhile中用lambda表达式，把prefix参数传入，既不用单独定义函数，又能支持动态前缀：

import itertools

def read_file(f, prefix):
    lines = []
    for line in itertools.dropwhile(lambda line: not line.startswith(prefix), f):
        lines.append(line.rstrip())
    return lines

lambda表达式在这里相当于一个匿名判断函数，直接把当前行和传入的前缀做比对，逻辑和原来的has_prefix完全一致，但更简洁。

2. 用列表推导式进一步简化 #

如果不需要逐行处理的中间步骤，可以直接用列表推导式一次性生成结果，代码更紧凑：

import itertools

def read_file(f, prefix):
    return [line.rstrip() for line in itertools.dropwhile(lambda line: not line.startswith(prefix), f)]

3. 用生成器函数（按需读取） #

如果文件很大，不想一次性加载所有行到内存，可以改成生成器函数，逐行返回结果：

import itertools

def read_file(f, prefix):
    for line in itertools.dropwhile(lambda line: not line.startswith(prefix), f):
        yield line.rstrip()

使用时直接遍历生成器即可，比如for line in read_file(open('file.txt'), 'real data'):，这样更节省内存。

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