如何使用通用 for 循环而不是列表扩展来简化这行代码？

Question

看看下面的一段代码：

def load_data_k(fname: str, yyy_index: int, **selection):
    selection_key_str = list(selection.keys())[0]
    selection_value_int = selection[selection_key_str]
    print(selection_value_int)
    i = 0
    file = open(fname)
    if "top_n_lines" in selection:
        lines = [next(file) for _ in range(selection_value_int)]

Answer 1

这个片段：

lines = [next(file) for _ in range(selection_value_int)]

扩展为：

lines = []
for _ in range(selection_value_int):
    lines.append(next(file))

然而，这并没有简化任何事情。

next(file)使用 File 对象的生成器行为因此加载一些行而不获取整个文件

Answer 2

你可以做

with open(fname) as file:
    lines = file.readlines()[:selection_value_int]