返回前 n 行 - Python

在 Python 中，我们可以利用以下几种方式返回前 n 行的数据。

方法一：用 for 循环输出前 n 行

def read_file_first_n_lines(file_path: str, n: int) -> list:
    result = []
    with open(file_path, 'r') as f:
        for i in range(0, n):
            line = f.readline()
            result.append(line.strip())
    return result

该函数的参数为文件路径和要返回的行数，函数会按照给定的行数读取文件中的数据，并将其保存到一个列表中返回。

方法二：使用列表表达式来输出前 n 行

def read_file_first_n_lines_v2(file_path: str, n: int) -> list:
    with open(file_path, 'r') as f:
        return [f.readline().strip() for i in range(0, n)]

该函数利用了列表表达式来实现读取前 n 行的数据的功能。

代码片段

如果你只需要返回前 n 行的数据，以上两种方法都是很好的选择。你也可以将这些代码片段复制到你的项目中，用它们来读取文件中的数据。

# 读取文件前 n 行的数据
def read_file_first_n_lines(file_path: str, n: int) -> list:
    result = []
    with open(file_path, 'r') as f:
        for i in range(0, n):
            line = f.readline()
            result.append(line.strip())
    return result

# 使用列表表达式输出前 n 行的数据
def read_file_first_n_lines_v2(file_path: str, n: int) -> list:
    with open(file_path, 'r') as f:
        return [f.readline().strip() for i in range(0, n)]